有关模型的定义、参数查看和训练;卷积层计算规则;模型修改的小技巧,请参考:https://blog.csdn.net/weixin_42468475/article/details/108628474?csdn_share_tail=%7B%22type%22%3A%22blog%22%2C%22rType%22%3A%22article%22%2C%22rId%22%3A%22108628474%22%2C%22source%22%3A%22weixin_42468475%22%7D

1 加载数据集

torchvision库里有很多数据集。我们这次用 CIFAR10 数据集,这是一个十分类的数据集。

import torch
import torchvision
<