下面的代码会将net网络的每层都进行冻结,如果需要冻结特定的某些层的话,可以根据该层的名字name来进行筛选,然后将requires_grad设置为false即可。
for name, p in net.named_parameters():
# 查看自己网络参数各层名称、数值
print(name, p)
p[1].requires_grad = False
下面的代码会将net网络的每层都进行冻结,如果需要冻结特定的某些层的话,可以根据该层的名字name来进行筛选,然后将requires_grad设置为false即可。
for name, p in net.named_parameters():
# 查看自己网络参数各层名称、数值
print(name, p)
p[1].requires_grad = False