Update autograd_tutorial.py

malfet · web-flow · commit 4c72e9d6c0b1 · 2022-08-29T13:19:18.000-07:00
diff --git a/beginner_source/blitz/autograd_tutorial.py b/beginner_source/blitz/autograd_tutorial.py
@@ -284,7 +284,7 @@
 
 from torch import nn, optim
 
-model = torchvision.models.resnet18(pretrained=True)
+model = resnet18(weights=ResNet18_Weights.DEFAULT)
 
 # Freeze all the parameters in the network
 for param in model.parameters():
