Sean

inconsistent implementation for saved .pth model

2

Hi, Thanks for sharing such a awesome work. As current implementation did not using model.module anymore. Using **model.load_state_dict(state_dict)** instead of **model.module.load_state_dict(state_dict)** works for all trained weights provided except **HR18-300W.pth** which...