I have trained a neural model and now I want to load it. Here is my code:
model = FlowNet()
model.load_state_dict(torch.load(‘/Users/hanjun/Desktop/model1000.pth’,map_location=lambda storage, loc:storage))
)But when I run it, the error makes me confused. How can I modify my code?