diff options
| author | zhang <zch921005@126.com> | 2022-09-13 23:21:25 +0800 |
|---|---|---|
| committer | zhang <zch921005@126.com> | 2022-09-13 23:21:25 +0800 |
| commit | 2fef28a07fcc9f43455b24188987f882220c05f2 (patch) | |
| tree | 75b3808c5daec81e25b4d27cee83432855deb8ad /dl/normalize/mnist_demo.py | |
| parent | b7e7dbcfac5bf8907d7d1e06ee6de2597a4c80f0 (diff) | |
bn train vs. eval
Diffstat (limited to 'dl/normalize/mnist_demo.py')
| -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 |