2021年1月26日
Pytorch固定部分参数(只训练部分层)(增量训练)（few-shot learn）（迁移学习）
摘要： 在迁移学习中我们经常会用到预训练模型，并在预训练模型的基础上添加额外层。训练时先将预训练层参数固定，只训练额外添加的部分。完了之后再全部训练微调。 在pytorch 固定部分参数训练时需要在优化器中施加过滤。 需要自己过滤 optimizer.SGD(filter(lambda p: p.requi 阅读全文
posted @ 2021-01-26 16:28 戒骄戒躁-沉淀积蓄 阅读(1610) 评论(0) 推荐(0)