实验四：迁移学习finetune：加载Resnet预训练的参数，修改Resnet的fc层，重新训练，完成4分类任务

一、准备工作
1. 模型准备

    模型下载：https://download.pytorch.org/models/resnet18-5c106cde.pth

    下载完后如下，有44.6M。

2. 数据集准备

   |----data

            |----pubu                                       #下载的数据集。

                       |----train

                                |----saopin                #136张图片。

                                |----wurenji               #136张图片。

                       |----val

                                |----saopin                #70张图片。

                                |----wurenji               #70张图片。

            |----resnet18-5c106cde.pth        #预训练的模型

   |----src

            |-----finetune_resnet18.py          #要写的代码

   |----tools                                              #通用的一些函数

            |----my_dataset.py                      #Dataset
 

标签：resnet18,训练,pth,finetune,Resnet,----,wurenji,70,图片
来源： https://blog.csdn.net/pengchengliu/article/details/117374886