2020.6.17 二

各网络结构通用模式

各个模块 import

hyperparameter 设置

数据加载 dataloader     import torch.utils.data as Data

train_loader = Data.DataLoader(dataset = train_data, batch_size = BATCH_SIZE, shuffle = True, num_workers = 0)

然后这个 train_data = torchvision.datasets.MNIST()