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

rhkr8521 · gogori6565 · rhkr8521 · commit 278d16c7b9f0 · 2024-09-09T17:26:06.000+09:00
Co-authored-by: Juhyeon Lee &lt;gogori6565@naver.com&gt;
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)
 
 
 ######################################################################

Original file line number	Diff line number	Diff line change
`@@ -124,7 +124,7 @@ def forward(self, x):`
`124`	`124`	`# 매치되도록 해 주면 됩니다.`
`125`	`125`	`#`
`126`	`126`
`127`		`-netB.load_state_dict(torch.load(PATH), strict=False)`
	`127`	`+netB.load_state_dict(torch.load(PATH, weights_only=True), strict=False)`
`128`	`128`
`129`	`129`
`130`	`130`	`######################################################################`