search

thohemp / 6DRepNet

Official Pytorch implementation of 6DRepNet: 6D Rotation representation for unconstrained head pose estimation.
MIT License
550 stars 72 forks source link

File "convert.py", line 35, in convert load_filtered_state_dict(model, saved_state_dict['model_state_dict']) KeyError: 'model_state_dict' #11

Closed APeiZou closed 2 years ago

APeiZou commented 2 years ago

@thohemp hello, python convert.py 6DRepNet_300W_LP_BIWI.pth 6DRepNet_300W_LP_BIWI_depoly.pth errors: File "convert.py", line 35, in convert load_filtered_state_dict(model, saved_state_dict['model_state_dict']) KeyError: 'model_state_dict'

thohemp commented 2 years ago

Hi, you try to deploy a model that is already deployed. The deploy script is for models that a generated by the training script.

https://github.com/thohemp/6DRepNet/blob/master/train.py#L193