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