Skip to content

Commit b13f072

Browse files
authored
remove deprecated argument
1 parent 6c8a3bf commit b13f072

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

beginner_source/fgsm_tutorial.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -192,7 +192,7 @@ def forward(self, x):
192192
model = Net().to(device)
193193

194194
# Load the pretrained model
195-
model.load_state_dict(torch.load(pretrained_model, weights_only=True, map_location=device))
195+
model.load_state_dict(torch.load(pretrained_model, map_location=device))
196196

197197
# Set the model in evaluation mode. In this case this is for the Dropout layers
198198
model.eval()

0 commit comments

Comments
 (0)