冻结神经网络的层可以参考这篇文章
冻结方法
冻结之后如何查看冻结成功没有呢?
用下面个代码即可
from __future__ import print_function, division
import torch
model1 = torch.load('model(frozen).pkl')
for k,v in model1.named_parameters():
print('{}: {}'.format(k, v.requires_grad))
这是运行结果,False表示冻结了,True表示没有冻结,我这里设置的是冻结1,2,3层,结果中前三层都为False,其他为True。


本文详细介绍了如何在PyTorch中冻结神经网络的特定层,通过示例代码展示了冻结过程及验证方法。运行结果表明,指定层已成功冻结,其余层保持可训练状态。

1166

被折叠的 条评论
为什么被折叠?



