ToTensor(),) # datasets.MNIST:是Pytorch的内置函数torchvision.datasets.MNIST,可以导入数据集 # train=True :读入的数据作为训练集 # transform:读入我们自己定义的数据预处理操作 # download=True:当我们的根目录(root)下没有数据集时,便自动下载 2 数据处理 如果是采用pickle.load加载数据,那么我们需要使用map...
一、数据预处理 (1)运行时所需库 importnumpyasnpimporttorchimporttorchvisionfromtorchimportnnfromtorch.utils.dataimportDataLoaderfromtorchvisionimportdatasetsimportmatplotlib.pyplotaspltimportos.path (2)选择合适的设备进行训练 device=torch.device("cuda"iftorch.cuda.is_available()else"cpu") (3) 构建数据集 ...