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