集的下载可以在kaggle官方下载，也可以使用代码进行下载。这个分类问题如果使用pre_train 的话，就相当于开外挂了，不过你也可以去看看结果到底如何。

（二）代码实现

import osimport torchimport torchvision from torch import nn from d2l import torch as d2l