We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 6c8a3bf commit b13f072Copy full SHA for b13f072
beginner_source/fgsm_tutorial.py
@@ -192,7 +192,7 @@ def forward(self, x):
192
model = Net().to(device)
193
194
# Load the pretrained model
195
-model.load_state_dict(torch.load(pretrained_model, weights_only=True, map_location=device))
+model.load_state_dict(torch.load(pretrained_model, map_location=device))
196
197
# Set the model in evaluation mode. In this case this is for the Dropout layers
198
model.eval()
0 commit comments