集的下載可以在kaggle官方下載，也可以使用代碼進行下載。這個分類問題如果使用pre_train 的話，就相當于開外掛了，不過你也可以去看看結(jié)果到底如何。

（二）代碼實現(xiàn)

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