PyTorch v2.4.0반영, pytorch/tutorials@ff7142e (PyTorchKorea/pytorch.kr#54)

Co-authored-by: Juhyeon Lee <[email protected]>
rhkr8521 · Sep 9, 2024 · 278d16c · 278d16c
1 parent 71feff4
commit 278d16c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/recipes_source/recipes/warmstarting_model_using_parameters_from_a_different_model.py b/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)
 
 
 ######################################################################