Pytorch模型保存和加载

保存模型:

torch.save(model, 'model.pth')

加载模型:

model = torch.load('model.pth')

Original: https://www.cnblogs.com/geeksongs/p/15861323.html
Author: Geeksongs
Title: Pytorch模型保存和加载

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/576915/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球