From 65b09f8b0c5d234bf5ae741eefbaa16a4884d42d Mon Sep 17 00:00:00 2001 From: aus612 Date: Thu, 23 Jul 2026 20:22:13 -0700 Subject: [PATCH] Fix ResNet18 preprocessing in Captum tutorial --- beginner_source/introyt/captumyt.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/beginner_source/introyt/captumyt.py b/beginner_source/introyt/captumyt.py index d08052caec..dc8b43bee2 100644 --- a/beginner_source/introyt/captumyt.py +++ b/beginner_source/introyt/captumyt.py @@ -143,7 +143,8 @@ # now. # -model = models.resnet18(weights='IMAGENET1K_V1') +weights = models.ResNet18_Weights.IMAGENET1K_V1 +model = models.resnet18(weights=weights) model = model.eval() @@ -166,9 +167,10 @@ # folder as well. # -# model expects 224x224 3-color image +# The model's weights resize the shorter side to 256 pixels before +# center-cropping to the expected 224x224 3-color image. transform = transforms.Compose([ - transforms.Resize(224), + transforms.Resize(256), transforms.CenterCrop(224), transforms.ToTensor() ])