test error load model

Question

test error load model

Closed this issue 7 years ago · 5 comments

TypeError: 'str' object is not callable

error line: test.py
state_dict = torch.load(save_path, map_location='cpu')

Answer 1 · 2018-04-12T01:42:09.000Z

It's hard to figure out where is the problem. Can you apply more details, like the error traceback, any changes from the code or your environment.

Answer 2 · 2018-04-12T01:47:21.000Z

state_dict = torch.load(save_path) #use this is OK

Answer 3 · 2018-04-12T01:48:54.000Z

in line utils.py load_network()

Answer 4 · 2018-04-12T04:02:10.000Z

Did you test on CPU? What's your pytorch version (torch.__version__)?

Try to replace utils.py line 38 state_dict = torch.load(save_path, map_location='cpu') with:
state_dict = torch.load(save_path, map_location=lambda storage, loc: storage)

Answer 5 · 2018-04-12T05:22:26.000Z

thx.
state_dict = torch.load(save_path, map_location=lambda storage, loc: storage)
and state_dict = torch.load(save_path) is ok