Add regression file for ImageNet batch

Signed-off-by: Fabrice Normandin <normandf@mila.quebec>
mila-iqia · Jun 19, 2024 · ac5f6e1 · ac5f6e1
1 parent 675a865
commit ac5f6e1
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/project/datamodules/datamodules_test/test_first_batch/imagenet.yaml b/project/datamodules/datamodules_test/test_first_batch/imagenet.yaml
@@ -0,0 +1,21 @@
+'0':
+  device: cpu
+  hash: -5237812157965531059
+  max: 2.64
+  mean: -0.067
+  min: -2.118
+  shape:
+  - 32
+  - 3
+  - 224
+  - 224
+  sum: -320363.531
+'1':
+  device: cpu
+  hash: 2821605806225718212
+  max: 982
+  mean: 511.219
+  min: 0
+  shape:
+  - 32
+  sum: 16359