preresnet83 resize to 32

mengxiangming · Oct 8, 2019 · d17b086 · d17b086
1 parent 0102ba4
commit d17b086
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/swag/models/preresnet.py b/swag/models/preresnet.py
@@ -205,6 +205,7 @@ class PreResNet83:
     kwargs = {"depth": 83}
     transform_train = transforms.Compose(
         [
+            transforms.Resize(32),
             transforms.RandomCrop(32, padding=4),
             transforms.RandomHorizontalFlip(),
             transforms.ToTensor(),
@@ -213,6 +214,7 @@ class PreResNet83:
     )
     transform_test = transforms.Compose(
         [
+            transforms.Resize(32),
             transforms.ToTensor(),
             transforms.Normalize((0.4914, 0.4822, 0.4465), (0.2023, 0.1994, 0.2010)),
         ]