qq_51941086的博客

使用全连接神经网络 我们接下来就是要预测类似下面的图片中的数字是多少 导入之后会用到的模块 import torch from torchvision import transforms, datasets from torch.utils.data import DataLoader import torch.nn.functional as F import torch.optim as optim 加载数据集，并且将数据集进行划分 batch_size = 64 transform = trans