From 2fef28a07fcc9f43455b24188987f882220c05f2 Mon Sep 17 00:00:00 2001
From: zhang <zch921005@126.com>
Date: Tue, 13 Sep 2022 23:21:25 +0800
Subject: bn train vs. eval

---
 dl/normalize/mnist_demo.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

(limited to 'dl')

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
-- 
cgit v1.2.3