beginner_source/basics/saveloadrun_tutorial.py 번역

beginner_source/basics/saveloadrun_tutorial.py

@@ -52,8 +52,13 @@
 
 ########################
 # 다음과 같이 모델을 불러올 수 있습니다:
+# 
+# `Saving and loading torch.nn.Modules <pytorch.org/docs/main/notes/serialization.html#saving-and-loading-torch-nn-modules>`__에서 설명한 것처럼,
+# ``state_dict``를 저장하는 것이 가장 좋은 방법으로 간주됩니다.
+# 하지만 아래에서는 ``weights_only=False``를 사용하는데,
+# 이는 모델을 로드하는 것을 포함하기 때문이며, ``torch.save``의 레거시 사용 사례입니다.
 
-model = torch.load('model.pth')
+model = torch.load('model.pth', weights_only=False),
 
 ########################
 # .. note:: 이 접근 방식은 Python `pickle <https://docs.python.org/3/library/pickle.html>`_ 모듈을 사용하여 모델을 직렬화(serialize)하므로, 모델을 불러올 때 실제 클래스 정의(definition)를 적용(rely on)합니다.

recipes_source/recipes/warmstarting_model_using_parameters_from_a_different_model.py

@@ -124,7 +124,7 @@ def forward(self, x):
 # 매치되도록 해 주면 됩니다.
 #
 
-netB.load_state_dict(torch.load(PATH), strict=False)
+netB.load_state_dict(torch.load(PATH, weights_only=True), strict=False)
 
 
 ######################################################################