Load_state_dict causes memory leak

[grahammorehead]

I’ll try that now and then post results here.