Open dk-hong opened 2 years ago
just add these after line 231: (as it is written for 256, it lacks mappings for the rest of the network) r"synthesis.b512.conv_last.weight", lambda: tf_params[f"synthesis/512x512/ToRGB/extraLayer/weight"].transpose(3, 2, 0, 1), r"synthesis.b512.conv_last.affine.weight", lambda: tf_params[f"synthesis/512x512/ToRGB/extraLayer/mod_weight"].transpose(), r"synthesis.b512.conv_last.affine.bias", lambda: tf_params[f"synthesis/512x512/ToRGB/extraLayer/mod_bias"] + 1, r"synthesis.b1024.conv_last.weight", lambda: tf_params[f"synthesis/1024x1024/ToRGB/extraLayer/weight"].transpose(3, 2, 0, 1), r"synthesis.b1024.conv_last.affine.weight", lambda: tf_params[f"synthesis/1024x1024/ToRGB/extraLayer/mod_weight"].transpose(), r"synthesis.b1024.conv_last.affine.bias", lambda: tf_params[f"synthesis/1024x1024/ToRGB/extraLayer/mod_bias"] + 1,
Cheers
I executed PyTorch_version/loader.py with ffhq-snapshot.pkl and it worked well.
However, I didn't work with ffhq-snapshot-1024.pkl.
How can I resolve this issue?
Error messages are below.
Thank you in advance.