KeyError: 'module name can\'t contain "."'

[HungryKonata]

How can I fix the problem?

[HungryKonata]

Thank you
With your help I have solved the problem

[sguphil]

Thank you
With your help I have solved the problem

how to fix this issue?

[kubitz]

The new version of Pytorch does not support module names with a "." in their name.

You can either use an older version, or patch the state_dict file, as well as the model definition.

You can patch it using this for example:

x = torch.load("netG.pth", map_location=lambda storage, loc: storage)
def change_key(state_dict, old, new):
    for _ in range(len(state_dict)):
        k, v = state_dict.popitem(False)
        state_dict[new if old == k else k] = v

change_key(x, "main.extra-layers-0.64.conv.weight", "main.extra-conv-0.weight")
change_key(x, "main.extra-layers-0.64.batchnorm.weight", "main.extra-bn-0.weight")
change_key(x, "main.extra-layers-0.64.batchnorm.bias", "main.extra-bn-0.bias")
change_key(x, "main.extra-layers-0.64.batchnorm.running_mean", "main.extra-bn-0.running_mean")
change_key(x, "main.extra-layers-0.64.batchnorm.running_var", "main.extra-bn-0.running_var")
change_key(x, "main.final_layer.deconv.weight", "main.final-deconv.weight")

[kubitz]

Also see issue #4