diff --git a/beginner_source/saving_loading_models.py b/beginner_source/saving_loading_models.py index 6c9b6b1fd77..0487830782d 100644 --- a/beginner_source/saving_loading_models.py +++ b/beginner_source/saving_loading_models.py @@ -153,7 +153,7 @@ # .. code:: python # # model = TheModelClass(*args, **kwargs) -# model.load_state_dict(torch.load(PATH), weights_only=True) +# model.load_state_dict(torch.load(PATH, weights_only=True)) # model.eval() # # .. note::