From 05ffcb261755e3b663e022ce36e2c41ff03eac21 Mon Sep 17 00:00:00 2001 From: Aditya Punetha <71438448+AdityaPunetha@users.noreply.github.com> Date: Mon, 29 Aug 2022 12:27:23 +0530 Subject: [PATCH 1/2] Update autograd_tutorial.py The parameter 'pretrained' is deprecated. Updated it to latest standard. --- beginner_source/blitz/autograd_tutorial.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/beginner_source/blitz/autograd_tutorial.py b/beginner_source/blitz/autograd_tutorial.py index d2c3824021d..e4d26db0b51 100644 --- a/beginner_source/blitz/autograd_tutorial.py +++ b/beginner_source/blitz/autograd_tutorial.py @@ -43,8 +43,9 @@ This tutorial work only on CPU and will not work on GPU (even if tensor are moved to CUDA). """ -import torch, torchvision -model = torchvision.models.resnet18(pretrained=True) +import torch +from torchvision.models import resnet18, ResNet18_Weights +model = resnet18(weights=ResNet18_Weights.DEFAULT) data = torch.rand(1, 3, 64, 64) labels = torch.rand(1, 1000) From 4c72e9d6c0b18efaaaa42a0c1ccc7b32cc8a0674 Mon Sep 17 00:00:00 2001 From: Nikita Shulga Date: Mon, 29 Aug 2022 13:19:18 -0700 Subject: [PATCH 2/2] Update autograd_tutorial.py --- beginner_source/blitz/autograd_tutorial.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/beginner_source/blitz/autograd_tutorial.py b/beginner_source/blitz/autograd_tutorial.py index e4d26db0b51..67336be7fa1 100644 --- 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():