在保存网络之前清理渐变的最简单方法是什么?

我保存一个nn.Module似乎好像不行,但通过挖掘代码我可以告诉它应该看起来像:
torch.save(net.state_dict(), 'net_name.pth')
这样用对不对?
但如何清洁/清除所有参数的之前保存的变量,就好像Torch 7中的clearState()
已邀请:
匿名用户

匿名用户

赞同来自:

你可以试试使用
torch.save(net.state_dict(), 'net_name.pth')

ptorch

赞同来自:

嗯嗯,可以了,谢谢你!

要回复问题请先登录注册