What’s the easiest way to clean gradients before saving a net?
<SCODE>torch.save(net.state_dict(), 'net_name.pth')
<ECODE> right?
pengsun
<SCODE>torch.save(net.state_dict(), 'net_name.pth')
<ECODE>
fmassa
pengsun