关于pytorch训练和验证图片的简单程序历程

这个是关于pytorch训练图片和验证图片数据集的简单程序历程

这个程序是学习b站上土堆的pytorch教程而写下的程序，欢迎大家学习土堆的有关教程，讲的挺好的。

import torchvision
from torch import nn
from torch.nn import Conv2d, MaxPool2d, Flatten, Linear
from torch.utils.data import DataLoader
from torch.utils.tensorboard import SummaryWriter

from model import *

train_data = torchvision.datasets.CIFAR10(root='dataset', train=True, transform=torchvision.transforms.ToTensor(),
                                          download=True)

test_data = torchvision.datasets.CIFAR10(root='dataset', train=False, transform=torchvision.transforms.ToTensor(),
                                         download=True)

train_data_dataloader = DataLoader(train_data, batch_size=64)
test_data_dataloader = DataLoader