摘要:
import torchimport torch.nn as nnimport torch.nn.functional as Fclass VGGbase(nn.Module): # 定义网络模型 def __int__(self): super(VGGbase, self).__init__() 阅读全文
摘要:
#自定义加载cifar10数据集from torchvision import transformsfrom torch.utils.data import DataLoader, Datasetimport osfrom PIL import Imageimport numpy as npimpo 阅读全文