模型搭建
模型搭建步骤:
![在这里插入图片描述](https://i-blog.csdnimg.cn/blog_migrate/19b80864c9d8931fa658884ac199ac50.png)
1.定义头文件
import torch as t
import torchvision as tv #数据处理,导入、预览
import torchvision.transforms as transforms #数据变换
from torch.autograd import Variable #梯度
import torch.nn as nn #实现神经网络相关功能的包
import torch.nn.functional as F
import torch.optim as optim #优化器
2.定义 dataloader
主要函数:
tv.datasets: 数据路径,是训练集还是测试集,是否需要下载,预处理
t.utils.data.Dataloader:导⼊数据,制定batch_size
tv.transform.Compose:数据预处理的操作集,如将数