包装产品做网站网站快速排名案例
1. 背景
冻结model中的cnn1层:
model.cnn1.requires_grad = False
运行后发现cnn1的参数仍然在更新
作为一个编程菜逼,我乍一看没毛病呀,凌晨1点的我越调越迷糊,终于最终还是找到了问题,还是基础不牢
2.原因
应使用
model.cnn1.weight.requires_grad = False
requires_grad是tensor的一个属性,而cnn1是一个卷积层,应注意神经网络中的参数指的是权重
离谱的是我上面的错误使用竟然没报错