error of pretrain models

grenlayk / deep-text-edit

Our implementation of Text Style Brush architecture.

MIT License

119 stars 27 forks source link

error of pretrain models #29

Open wushilian opened 1 year ago

wushilian commented 1 year ago

when train the model ,there is a error:

RuntimeError: Error(s) in loading state_dict for DataParallel: Missing key(s) in state_dict: "module.Transformation.GridGenerator.inv_delta_C", "module.Transformation.GridGenerator.P_hat".