add imagenette and *args to imagedataset; fix default normalization t…

…o None; update mnnist config with normalization factors instead
slegroux · Jan 5, 2025 · 671ad09 · 671ad09
1 parent 58f40fa
commit 671ad09
Show file tree

Hide file tree

Showing 3 changed files with 119 additions and 94 deletions.
diff --git a/config/data/image/mnist.yaml b/config/data/image/mnist.yaml
@@ -7,4 +7,7 @@ num_workers: 0
 pin_memory: False
 persistent_workers: False
 transforms: 
-  _target_: torchvision.transforms.ToTensor
+  - _target_: torchvision.transforms.ToTensor
+  - _target_: transforms.Normalize
+    mean: [0.1307,], 
+    std: [0.3081,]