Hi
I think the method you are using of considering
torch.save(net.state_dict(), 'net_name.pth')

is the current advised way of doing it, and it will only save the weights and not the gradients.
Indeed, since https://github.com/pytorch/pytorch/pull/451 44 the state_dict function only returns a tensor, and not the corresponding Variable.