摘要:
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小白
阅读(387)
评论(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小白
阅读(332)
评论(0)
推荐(0)