关于pytorch中的模型保存

作者: Songger | 来源:发表于2019-04-28 22:28 被阅读0次

//推荐使用的方式
t.save(net.state_dict(), 'net.pth')
//下面这种方式会依赖路径（在保存的key中会有路径名），不建议使用
t.save(net, 'net.pth')

网友评论

本文标题：关于pytorch中的模型保存

本文链接：https://www.haomeiwen.com/subject/qfusnqtx.html

备案信息：桂公网安备 45052102000051号 · 桂ICP备13007215号-3

本站所收录作品、热点评论等信息部分来源互联网，目的只是为了系统归纳学习和传递资讯

所有作品版权归原创作者所有，与本站立场无关，如不慎侵犯了你的权益，请联系我们告知，我们将做删除处理！