Pytorch loading model error resolution

One 、 Problem description

Evaluate the code downloaded from the network , When using automatic model loading “'xxx object has no attribute 'copy'” Error prompt for .

Two 、 Problem solving

1. This problem arises when saving the model torch.save(mode,"xxxx.pth") command , Instead of torch.save(model.state_dict(),"xxx.pth"), The resulting saved model is not a dictionary type .
2. We can take the following steps to save it as a dictionary type and then reload it .

pre_weights = torch.load("model.pth")  # Load model 
torch.save(pre_weights.state_dict(),"/tmp/1.pth") # Save it as /tmp/1.pth file 
model.load_state_dict(torch.load("/tmp/1.pth")) # Reload the model file