摘要:
(一)神经网络的骨架 nn.Module import torch from torch import nn class Tudui(nn.Module): def __init__(self): super().__init__() def forward(self, input): output 阅读全文
posted @ 2022-08-01 10:15
helloWorldhelloWorld
阅读(65)
评论(0)
推荐(0)
摘要:
batch_size:有多少张 shuffle=True:顺序不打乱 num_workers: 进程数 drop_last:最后不够64张是否舍去 import torchvision from torch.utils.data import DataLoader # 1 from torch.ut 阅读全文
posted @ 2022-08-01 09:37
helloWorldhelloWorld
阅读(34)
评论(0)
推荐(0)