I see.
besides the breaking issue, the loading code is fine?
I mean we dont need the 2nd and third element of torch.load output, since they are criterion and optimizer, right? all parameters should be in model?