如何查看神经网络中的层是否被冻结

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

冻结神经网络的层可以参考这篇文章
冻结方法

冻结之后如何查看冻结成功没有呢?
用下面个代码即可

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。
在这里插入图片描述
在这里插入图片描述

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值