(深度学习记录）第P2周：CIFAR10彩色图片识别

•  🍨 本文为🔗365天深度学习训练营 中的学习记录博客
• 🍖 原作者：K同学啊​​​​​

我的环境

• 电脑系统：Windows 11
• 编译器：Pycharm
• 深度学习环境：Pytorch

代码及结果

import torch
import torch.nn as nn
import matplotlib.pyplot as plt
import torchvision


device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

train_ds = torchvision.datasets.CIFAR10('data',
                                      train=True,
                                      transform=torchvision.transforms.ToTensor(), # 将
                                      download=True)

test_ds  = torchvision.datasets.CIFAR10('data',
                                      train=False,
                                      transform=torchvision.transforms.ToTensor(), # 将
                                      download=True)
batch_size = 32

train_dl = torch.