Qiulin-W / SAFA

Official Pytorch Implementation of 3DV2021 paper: SAFA: Structure Aware Face Animation.
Other
177 stars 29 forks source link

About checkpoint #8

Closed winterplayer closed 2 years ago

winterplayer commented 2 years ago

I download SAFA model checkpoint from Google Drive,and use data.pkl to the checkpoint.But there are some errors. File "/home/mpl/anaconda3/envs/py37/lib/python3.7/site-packages/torch/serialization.py", line 764, in _legacy_load magic_number = pickle_module.load(f, **pickle_load_args) _pickle.UnpicklingError: A load persistent id instruction was encountered, but no persistent_load function was specified.

VisionaryMind commented 2 years ago

@winterplayer , this error will be seen if you extract the serialized TAR file and access the PKL directly. The checkpoint should reference the TAR directly: ./ckpt/final_3DV.tar.

winterplayer commented 2 years ago

Thank you!The error is disappeared.