PyTorch error loading saved nn.Module: object has no attribute 'to'

Question

I am using PyTorch 0.4.

I defined a PyTorch MyModel by inheriting from nn.Module, and saved an instance of it by calling

torch.save(my_model, my_path)

Then, when loading it again with torch.load(my_path), my program crashed with the following error:

AttributeError: 'MyModel' object has no attribute 'to'

But my program was able to run it in previous stages. What did go wrong?

fr_andres · Accepted Answer · 2018-07-12 13:13:02Z

4

I already found it out, and just wanted to quickly post about it since google didn't give an obvious clue.

It turned out that, although I saved the model from a computer with 0.4, I was trying to load it from a different computer that still had an older (<0.4) PyTorch version installed. pip install --upgrade torch fixed it.

I found it out because the my_model.train()and .eval() methods were indeed working, so I remembered that the .to() method was introduced in 0.4. Useful references:

answered Jul 12, 2018 at 13:13

fr_andres

6,8972 gold badges39 silver badges49 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

PyTorch error loading saved nn.Module: object has no attribute 'to'

1 Answer 1

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related