pytorch 加载模型 模型大小测试速度

直接加载整个模型

Pytorch保存和加载整个模型：

  

   

    

     

    
    

     

           torch.save(model, 'model.pth')
     
    
   

    

     

    
    

     

           model = torch.load('model.pth')
     
    
  
 

Pytorch保存和加载预训练模型参数：

  

   

    

     

    
    

     

           torch.save(model.state_dict(), 'params.pth')
     
    
   

    

     

    
    

     

           model.load_state_dict(torch.load('params.pth'))
     
    
  
 

网络加层了,但是预训练没有，加载模型方法：

model.load_state_dict(state_dict,strict=False)
 

网络是cpu版本时：

  

   

    

     

    
    

     

        model_path = "../best_mode.pth"
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

       a_net.load_state_dict(torch.load(model_path, map_location=torch.device('cpu')))
     
    
  
 

测试并获取模型大小： 

  

   

    

     

    
    

     

       size=112
     
    
   

    

     

    
    

     
 import torch
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

       model = CNNModel(scale=1, num_classes=100,width=size,height=size)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

       model.cuda
     
    
  
 

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

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