一、头文件 import torchvision from torch.utils.tensorboard import SummaryWriter 二、训练集和数据集的设置 dataset_transform=torchvision.transforms.Compose( [torchvision.transforms.ToTensor()] ) train_set=torchvisi