1、数据集制作

 （1）将从kaggle下载的数据集中的train目录下的图片分类为cat和dog文件夹，每个文件夹下12500张图片

（2）使用教程1的方法进行数据的划分

 （1）数据集的划分_chencaw的博客-CSDN博客

2、使用resnet18的预训练模型迁移训练

（1）使用了dqtm查看进度

参考了tqdm介绍及常用方法_GZKPeng的博客-CSDN博客_tqdm函数

（2）使用summary()清晰打印网络结果

参考了

PyTorch 打印网络模型结构-pudn.com

1）安装

pip install torchinfo
或者
conda install -c conda-forge torchinfo