我是pytorch的新手,我想知道如何显示损失和准确度的图表,以及我应该如何存储这些值,因为我知道我正在使用CIFAR10应用cnn模型进行图像分类。 下面是我目前的实现: def train(num_epochs,optimizer,criterion,model):
for epoch in range(num_epochs):
for i, (images, labels) in enumerate(trainloader):
# origin shape: [4, 3, 32, 32] = 4, 3, 1