Has there been a follow-up or a solution to this issue? I am experiencing similar troubles with save and resume of states on pytorch 0.4.1.
I am willing to share code (a private gitlab repo) and the data (public dataset) in order to help reproduce the issue. Please advise.
Thanks and regards,
Bhargava