【文档学习】PyTorch——模型的保存和加载

本系列记录了博主学习PyTorch过程中的笔记,官方tutorial。 本文介绍了模型的保存和加载,官方示例。更新于2019.03.21。 这个文档提供了关于保存和加载PyTorch模型的一系列使用案例。可以全看,也可以根据需要选择。 在保存和加载模块时,需要了解下面三个核心函数: torch.save:保存一个序列化(serialized)的目标到磁盘。函数使用了Python的pickle程序用
相关文章
相关标签/搜索