Trying to Save and Load resnet50 net

1

ERROR PyTorch 0.1.11

2

What’s object? You didn’t show that in your snippet.

You don’t need to implement torch.load. Just call it with the file name. Also, note that we recommend using state_dict() and load_state_dict() for serialization.

3

I edited the code above to include all of the model, including training code.

It looks like I am using torch.save incorrectly. What is object supposed to refer to?
I am still unsure of what would be the correct way to use torch.save to save it and to load it, while keeping the resnet50 architecture?

Would this be correct?
torch.save(net.new, "/home/ubuntu/pics/test3.pth")

Hmm, ok I will look into state_dict()
It looks like the documentation on state_dict() is a bit sparse. Do you have any pointers on how to edit the code above to save and load using state_dict and load_state_dict()?

Thanks again

4

You still haven’t included the definition of object in the snippet. I have no idea what it is

5

Hi again,

I managed to figure it out, thanks for the help!