Advertisement

基于预训练的模型固定了部分层结构

阅读量:

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)

还没有任何评论哟~