the pretrained weights dont match generator of model

vincent-thevenin / Realistic-Neural-Talking-Head-Models

My implementation of Few-Shot Adversarial Learning of Realistic Neural Talking Head Models (Egor Zakharov et al.).

GNU General Public License v3.0

828 stars 195 forks source link

the pretrained weights dont match generator of model #59

Open Sregaler opened 3 years ago

Sregaler commented 3 years ago

RuntimeError: Error(s) in loading state_dict for Generator: Missing key(s) in state_dict: "conv2d.weight", "conv2d.bias". Unexpected key(s) in state_dict: "resDown5.conv_l1.bias", …………

Sregaler commented 3 years ago

how to use the master branch instead of default?

cindy-kika commented 3 years ago

I also met this problem. How did you solve it

Sregaler commented 3 years ago

I gave up the author's pretrained weights.I retrained in my own database, and now the meta learning stage works well.

kangyeolk commented 3 years ago

Check out this issue. There are two branches in this repository

#46