diff options
Diffstat (limited to 'dl')
| -rw-r--r-- | dl/normalize/mnist_demo.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/dl/normalize/mnist_demo.py b/dl/normalize/mnist_demo.py index dc9e00c..a835542 100644 --- a/dl/normalize/mnist_demo.py +++ b/dl/normalize/mnist_demo.py @@ -11,7 +11,8 @@ import torch # timm.data.IMAGENET_DEFAULT_STD: (0.229, 0.224, 0.225) transform = transforms.Compose([ transforms.ToTensor(), - transforms.Normalize(mean=[0.1307], std=[0.3081]) + transforms.Normalize(mean=[0.1307], std=[0.3081]), + transforms.Resize() ]) # MNIST dataset |