search

grey-eye / talking-heads

Our implementation of "Few-Shot Adversarial Learning of Realistic Neural Talking Head Models" (Egor Zakharov et al.)
GNU General Public License v3.0
593 stars 109 forks source link

vggface checkpoint import error #18

Closed akshay951228 closed 5 years ago

akshay951228 commented 5 years ago

Could you tell me which part was wrong? I am using VGGFace 16 at this moment. Did I download the wrong file or should I change which part of code? Thank you. File "run.py", line 283, in main() File "run.py", line 279, in main raise e File "run.py", line 265, in main continue_id=args.continue_id, File "run.py", line 67, in meta_train criterion_E_G = network.LossEG(device, feed_forward=True) File "D:\talking-heads\network\loss.py", line 17, in init self.VGG_FACE_AC = VGG_Activations(vgg_face(pretrained=True), [1, 6, 11, 18, 25]).type(self.dtype) File "D:\talking-heads\network\vgg.py", line 36, in vgg_face model.load_state_dict(vgg_face_state_dict()) File "D:\talking-heads\network\vgg.py", line 43, in vgg_face_state_dict 'features.0.weight': default['conv1_1.weight'], KeyError: 'conv1_1.weight' checkpoint file name :vgg_m_face_bn_fer_dag.pth from this link http://www.robots.ox.ac.uk/~albanie/pytorch-models.html

keishatsai commented 5 years ago

I think that you download the wrong file. I did this before. read the download file carefully.