pytorch 加载模型：

1.直接加载网络

import torch

pthfile = r'E:\anaconda\app\envs\luo\Lib\site-packages\torchvision\models\squeezenet1_1.pth'

net = torch.load(pthfile)

print(net)

  

   

    

     

    
    

     

      import  torch
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      def remove_prefix(state_dict, prefix):
     
    
   

    

     

    
    

     
 ''' Old style model is stored with all names of parameters
     
    
   

    

     

    
    

     

       share common prefix 'module.' '''
     
    
   

    

     

    
    

     
 # logger.info('remove prefix \'{}\''.format(prefix))
     
    
   

    

     

    
    

     

       f = lambda x: x.split(prefix, 1)[-1] if x.startswith(prefix) else x
     
    
   

    

     

    
    

     
 return {f(key): value for key, value in state_dict.items()}
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      def load_model(model, pretrained_path, load_to_cpu):
     
    
   

    

     

    
    

     

       print('Loading pretrained model from {}'.format(pretrained_path))
     
    
   

    

     

    
    

     
 if load_to_cpu:
     
    
   

    

     

    
    

     

       pretrained_dict = torch.load(pretrained_path, map_location=lambda storage, loc: storage)
     
    
   

    

     

    
    

     
 else:
     
    
   

    

     

    
    

     

       device = torch.cuda.current_device()
     
    
   

    

     

    
    

     

       pretrained_dict = 
     
    
  
 

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

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