diff --git a/beginner_source/basics/saveloadrun_tutorial.py b/beginner_source/basics/saveloadrun_tutorial.py index e80d32a6ea..8683335baf 100644 --- a/beginner_source/basics/saveloadrun_tutorial.py +++ b/beginner_source/basics/saveloadrun_tutorial.py @@ -63,6 +63,7 @@ # model, which is a legacy use case for ``torch.save``. model = torch.load('model.pth', weights_only=False) +model.eval() ######################## # .. note:: This approach uses Python `pickle `_ module when serializing the model, thus it relies on the actual class definition to be available when loading the model.