- 博客(2)
- 收藏
- 关注
原创 PYTORCH 实现CNN,迁移学习,残差网络
1. pytorch 实现CNN import torch import torch.nn as nn import torch.optim as optim import torch.nn.functional as F from torchvision import datasets, transforms import matplotlib.pyplot as plt import numpy as np '''step1 定义参数''' # 定义超参数 input_size = 28 #
2021-12-11 16:00:18
932
原创 pytorch入门
1.数据结构 使用pytorch需要将数据转换成模型需要的数据格式,数据格式包括单个数据对应的scalar格式,表示特征的vector格式,对应矩阵的matrix格式,对应多维数据的n-ddimensional tensor格式。数据转化有以下的方式。 '''使用tnesor进行数据转换''' import torch torch.tensor(input_features[start:end] '''使用TensorDataset,DataLoader 进行数据格式修改''' f...
2021-12-03 19:36:10
1761
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人
RSS订阅