基于预训练的模型固定了部分层结构
发布时间
阅读量:
阅读量
1、可以先看看各层的信息
"""
函数说明: 打印模型里面各层的信息,陈20221012
:param model: 加载的模型
:return:
"""
def look_layer(model):
cnt = 0
print("看看模型的param.requires_grad和个数1:")
for name, param in model.named_parameters():
cnt += 1
print("now cnt ",cnt,name,param.requires_grad,param.shape)
print(cnt)

2、冻结神经网络参数不训练的参数
param.requires_grad = False #将该参数设为False后,模型在训练过程中将不再对其进行更新
全部评论 (0)
还没有任何评论哟~
