Pytorch深度模型保存和加载

Pytorch保存模型的两种方式：

1 模型结构和模型参数都保存下来

优点：不需要预初始化模型，直接加载，就可以获得模型和参数，只要把数据放进去就可以进行测试

缺点：每个模型存储的文件比较大，一般的会有几个G。

2 只存储模型的参数，不存储模型结构

优点：存储文件较小

前提： 需要预初始化模型