2026/6/26 22:45:19
import torch
import torchvision
from PIL import Image
from torch import nn
from torch.nn import Sequential, Conv2d, MaxPool2d, Flatten, Linear# 1. 网络结构(必须和训练时完全一致)
class Tudui(nn.Module):def __init__(self):super(Tudui,…