I try to load pretrained model by:
saved = torch.save(model.state_dict())
model.load_state_dict(torch.load(saved))
but the ParameterList([Parameter(Tensor1), Parameter(Tensor2) …]) of the model is not correctly loaded.
So, what is the correct method to load pretrained weights for nn.ParameterList()?
pytorch 1.6.0