关于pytorch中的模型保存

2019-04-28  本文已影响0人  Songger
//推荐使用的方式
t.save(net.state_dict(), 'net.pth')
//下面这种方式会依赖路径(在保存的key中会有路径名),不建议使用
t.save(net, 'net.pth')
上一篇 下一篇

猜你喜欢

热点阅读