1、pytorch载入预训练模型后,只想训练个别层怎么办?
https://blog.csdn.net/qq_34351621/article/details/79967463
如果载入的这些参数中,有些参数不要求被更新,即固定不变,不参与训练,需要手动设置这些参数的梯度属性为Fasle,并且在optimizer传参时筛选掉这些参数。
迁移学习--冻结模型
最新推荐文章于 2024-05-18 08:38:02 发布
1、pytorch载入预训练模型后,只想训练个别层怎么办?
https://blog.csdn.net/qq_34351621/article/details/79967463
如果载入的这些参数中,有些参数不要求被更新,即固定不变,不参与训练,需要手动设置这些参数的梯度属性为Fasle,并且在optimizer传参时筛选掉这些参数。