pytorch 打印模型参数

打印参数：

  

   

    

     

    
    

     

      for name, param in vgg16.state_dict():
     
    
   

    

     

    
    

     
    print(name, ' ', param.size())
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      for name, param in vgg16.named_parameters():
     
    
   

    

     

    
    

     
	print(name, ' ', param.size())
     
    
  
 

  

   

    

     

    
    

     

      fc = nn.Linear(3, 5)
     
    
   

    

     

    
    

     

      params = list(fc.named_parameters())
     
    
   

    

     

    
    

     

      print(params.__len__())
     
    
   

    

     

    
    

     

      print(params[0])
     
    
   

    

     

    
    

     

      print(params[1])
     
    
  
 

文章来源: blog.csdn.net，作者：AI视觉网奇，版权归原作者所有，如需转载，请联系作者。

原文链接：blog.csdn.net/jacke121/article/details/121349883