autonise/CRAFT-Remade

Error on loading final_model.pkl

Closed this issue · 1 comments

devloper13 commented

I am getting this 'RuntimeError: Error(s) in loading state_dict for DataParallelModel' while loading final_model.pkl

What could this be?

mayank-git-hub commented

@devloper13 sorry for the late reply. This error is usually is seen in the following cases -

  1. You have made changes in the model architecture after saving the model file.
  2. You saved the model as a DataParallel object but are trying to read it without wrapping using DataParallel