No module named '__builtin__' while running torch.load(f"{SaveFolder}/model.pt", map_location = torch.device('cpu'))

Viewed 17

I'm trying to load a previously trained model using the following:

model = torch.load(f"{SaveFolder}/model.pt", map_location = torch.device('cpu'))

SaveFolder has the correct path of the previously saved fully trained model. And this is the error I'm getting:

enter image description here

1 Answers

Try: pip install future and see if that clears the issue up.

Related