From 9b3c476b67f161c370477cef57a202731fbcc4ea Mon Sep 17 00:00:00 2001 From: Berin Aniesh Date: Fri, 30 Aug 2024 11:08:31 +0530 Subject: [PATCH] fix typo in loading model from disk code --- beginner_source/saving_loading_models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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::