迁移学习--冻结模型

1、pytorch载入预训练模型后,只想训练个别层怎么办?
https://blog.csdn.net/qq_34351621/article/details/79967463
如果载入的这些参数中,有些参数不要求被更新,即固定不变,不参与训练,需要手动设置这些参数的梯度属性为Fasle,并且在optimizer传参时筛选掉这些参数。

你可能感兴趣的:(Pytorch,迁移学习)