when i was trying to the load the model with pre trained weights I was getting the following error
RuntimeError: Error(s) in loading state_dict for VAEGen:
Missing key(s) in state_dict: “enc.model.0.conv.weight”, “enc.model.0.conv.bias”, “enc.model.1.conv.weight”, “enc.model.1.conv.bias”, “enc.model.2.conv.weight”, “enc.model.2.conv.bias”, “enc.model.3.model.0.model.0.conv.weight”, “enc.model.3.model.0.model.0.conv.bias”, “enc.model.3.model.0.model.1.conv.weight”, “enc.model.3.model.0.model.1.conv.bias”, “enc.model.3.model.1.model.0.conv.weight”, “enc.model.3.model.1.model.0.conv.bias”, “enc.model.3.model.1.model.1.conv.weight”, “enc.model.3.model.1.model.1.conv.bias”, “enc.model.3.model.2.model.0.conv.weight”, “enc.model.3.model.2.model.0.conv.bias”, “enc.model.3.model.2.model.1.conv.weight”, “enc.model.3.model.2.model.1.conv.bias”, “enc.model.3.model.3.model.0.conv.weight”, “enc.model.3.model.3.model.0.conv.bias”, “enc.model.3.model.3.model.1.conv.weight”, “enc.model.3.model.3.model.1.conv.bias”, “dec.model.0.model.0.model.0.conv.weight”, “dec.model.0.model.0.model.0.conv.bias”, “dec.model.0.model.0.model.1.conv.weight”, “dec.model.0.model.0.model.1.conv.bias”, “dec.model.0.model.1.model.0.conv.weight”, “dec.model.0.model.1.model.0.conv.bias”, “dec.model.0.model.1.model.1.conv.weight”, “dec.model.0.model.1.model.1.conv.bias”, “dec.model.0.model.2.model.0.conv.weight”, “dec.model.0.model.2.model.0.conv.bias”, “dec.model.0.model.2.model.1.conv.weight”, “dec.model.0.model.2.model.1.conv.bias”, “dec.model.0.model.3.model.0.conv.weight”, “dec.model.0.model.3.model.0.conv.bias”, “dec.model.0.model.3.model.1.conv.weight”, “dec.model.0.model.3.model.1.conv.bias”, “dec.model.2.norm.gamma”, “dec.model.2.norm.beta”, “dec.model.2.conv.weight”, “dec.model.2.conv.bias”, “dec.model.4.norm.gamma”, “dec.model.4.norm.beta”, “dec.model.4.conv.weight”, “dec.model.4.conv.bias”, “dec.model.5.conv.weight”, “dec.model.5.conv.bias”.
Unexpected key(s) in state_dict: “a”, “b”.
I don’t know what could be the reason