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