pytorch加载和保存模型
摘要:
在模型完成训练后,我们需要将训练好的模型保存为一个文件供测试使用,或者因为一些原因我们需要继续之前的状态训练之前保存的模型,那么如何在PyTorch中保存和恢复模型呢? 方法一(推荐): 第一种方法也是官方推荐的方法,只保存和恢复模型中的参数。 保存 torch.save(the_model.sta 阅读全文
posted @ 2017-10-12 16:16 nkh 阅读(32181) 评论(0) 推荐(0) 编辑