pytorch 模型加载与保存

翻译自https://pytorch.org/tutorials/beginner/saving_loading_models.html 保存和与加载模型,有三个核心函数需要熟悉: torch.save: 保存一个序列化的对象至硬盘,。该函数使用了Python的pickle包用于序列化。模型、张量和各种对象的字典都可以使用该函数保存; torch.load:使用pickle的反序列化功能将序列化的
相关文章
相关标签/搜索