Add other (?) missing ImageNet regression files

Signed-off-by: Fabrice Normandin <[email protected]>
mila-iqia · Nov 22, 2024 · 72a77ff · 72a77ff
1 parent 8544e4a
commit 72a77ff
Show file tree

Hide file tree

Showing 2 changed files with 40 additions and 0 deletions.
diff --git a/...ifier_test/test_forward_pass_is_reproducible/cuda/resnet18_imagenet_image_classifier.yaml b/...ifier_test/test_forward_pass_is_reproducible/cuda/resnet18_imagenet_image_classifier.yaml
@@ -0,0 +1,20 @@
+input:
+  device: cuda:0
+  max: '2.640e+00'
+  mean: '-6.663e-02'
+  min: '-2.118e+00'
+  shape:
+  - 64
+  - 3
+  - 224
+  - 224
+  sum: '-6.419e+05'
+out:
+  device: cuda:0
+  max: '2.662e+00'
+  mean: '1.718e-03'
+  min: '-2.466e+00'
+  shape:
+  - 64
+  - 1000
+  sum: '1.099e+02'
diff --git a/...ifier_test/test_forward_pass_is_reproducible/cuda/resnet50_imagenet_image_classifier.yaml b/...ifier_test/test_forward_pass_is_reproducible/cuda/resnet50_imagenet_image_classifier.yaml
@@ -0,0 +1,20 @@
+input:
+  device: cuda:0
+  max: '2.640e+00'
+  mean: '-6.663e-02'
+  min: '-2.118e+00'
+  shape:
+  - 64
+  - 3
+  - 224
+  - 224
+  sum: '-6.419e+05'
+out:
+  device: cuda:0
+  max: '6.429e+00'
+  mean: '5.057e-04'
+  min: '-5.682e+00'
+  shape:
+  - 64
+  - 1000
+  sum: '3.237e+01'