1.在model.py搭建神经网络。 # 搭建神经网络 10分类网络。 import torch from torch import nn class net(nn.Module): def __init__(self): super(net, self).__init__() self.model = nn.Sequential( # 卷积 nn.Conv2d(in_channels=3, out_channels=32, kernel_size=5, stride=1, padding=2), # 最大池化 nn.MaxPool2d(kernel_size=2), # 卷积 nn.Conv2d(in_channels=32, out_channels=32, kernel_size=5, stride=1, padding=2), # 最大池化 nn.MaxPool2d(kernel_size=2), # 卷积 nn.Conv2d(in_channels=32, out_channels=64, kernel_size=5, stride=1, padding=2), #...