Pytorch深度模型保存和加载

Pytorch保存模型的两种方式：

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

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

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


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

优点：存储文件较小

前提： 需要预初始化模型


©著作权归作者所有,转载或内容合作请联系作者
平台声明：文章内容（如有图片或视频亦包括在内）由作者上传并发布，文章内容仅代表作者本人观点，简书系信息发布平台，仅提供信息存储服务。

推荐阅读更多精彩内容