Save model and use in pytorch (without fastai)?

[muellerzr (Zachary Mueller)]

I think a better example may be here: Loading pretrained weights that are not from ImageNet

What you should also do is on new_state_dict (when we assign it) also tag in the [‘model’] to the end.