-
Notifications
You must be signed in to change notification settings - Fork 78
Error while converting weights #5
Description
Hi I tried to convert the weights from the original repository by Nvidialabs according to convert.py, and there have been errors loading the state dictionary. Please let me know if there is any extra information you'd like to know.
My Tensorflow version is 1.10.0
>>> gen.load_state_dict(param_dict, strict=False)
_IncompatibleKeys(missing_keys=['g_synthesis.blocks.0.conv0_up.intermediate.kernel', 'g_synthesis.blocks.1.conv0_up.intermediate.kernel', 'g_synthesis.blocks.2.conv0_up.intermediate.kernel', 'g_synthesis.blocks.3.conv0_up.intermediate.kernel', 'g_synthesis.blocks.4.conv0_up.intermediate.kernel', 'g_synthesis.blocks.5.conv0_up.intermediate.kernel', 'g_synthesis.blocks.6.conv0_up.intermediate.kernel', 'g_synthesis.blocks.7.conv0_up.intermediate.kernel', 'g_synthesis.to_rgb.0.weight', 'g_synthesis.to_rgb.0.bias', 'g_synthesis.to_rgb.1.weight', 'g_synthesis.to_rgb.1.bias', 'g_synthesis.to_rgb.2.weight', 'g_synthesis.to_rgb.2.bias', 'g_synthesis.to_rgb.3.weight', 'g_synthesis.to_rgb.3.bias', 'g_synthesis.to_rgb.4.weight', 'g_synthesis.to_rgb.4.bias', 'g_synthesis.to_rgb.5.weight', 'g_synthesis.to_rgb.5.bias', 'g_synthesis.to_rgb.6.weight', 'g_synthesis.to_rgb.6.bias', 'g_synthesis.to_rgb.7.weight', 'g_synthesis.to_rgb.7.bias'], unexpected_keys=['truncation.avg_latent'])