Update fine-tuning.md

astonzhang · web-flow · commit b2c86d275189 · 2022-12-03T17:49:24.000-08:00
diff --git a/chapter_computer-vision/fine-tuning.md b/chapter_computer-vision/fine-tuning.md
@@ -159,7 +159,7 @@ train_augs = torchvision.transforms.Compose([
     normalize])
 
 test_augs = torchvision.transforms.Compose([
-    torchvision.transforms.Resize(256),
+    torchvision.transforms.Resize(256, 256),
     torchvision.transforms.CenterCrop(224),
     torchvision.transforms.ToTensor(),
     normalize])
