各网络结构通用模式
各个模块 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()