Transfer Learning & Model Finetune
Transfer Learning:机器学习分支,研究源域(source domain)的知识如何应用到目标域(target domain)
PyTorch中的Finetune
Model Finetune:模型的迁移学习
模型微调步骤:
1. 获取预训练模型参数
2. 加载模型(load_state_dict)
3. 修改输出层
模型微调训练方法:
1. 固定预训练的参数(requires_grad =False;lr=0)
2. Features Extractor较小学习率(params_group)