pytorch保存、加载和解析模型权重

1、模型保存和加载

         主要有两种情况：一是仅保存参数，二是保存参数及模型结构。

保存参数：

         torch.save(net.state_dict())

加载参数（加载参数前需要先实例化模型）：

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

         net.load_state_dict(param)</