摘要: import torchimport torch.nn as nnimport torch.nn.functional as Fclass VGGbase(nn.Module): # 定义网络模型 def __int__(self): super(VGGbase, self).__init__() 阅读全文
posted @ 2021-08-22 20:09 编程coding小白 阅读(350) 评论(0) 推荐(0) 编辑
摘要: #自定义加载cifar10数据集from torchvision import transformsfrom torch.utils.data import DataLoader, Datasetimport osfrom PIL import Imageimport numpy as npimpo 阅读全文
posted @ 2021-08-22 20:00 编程coding小白 阅读(297) 评论(0) 推荐(0) 编辑