Update regression files

Signed-off-by: Fabrice Normandin <[email protected]>
mila-iqia · Nov 20, 2024 · e468c07 · e468c07
1 parent 0fb4824
commit e468c07
Show file tree

Hide file tree

Showing 5 changed files with 542 additions and 146 deletions.
diff --git a/...ssifier_test/test_backward_pass_is_reproducible/cifar10_jax_cnn_jax_image_classifier.yaml b/...ssifier_test/test_backward_pass_is_reproducible/cifar10_jax_cnn_jax_image_classifier.yaml
@@ -75,20 +75,20 @@ grads.network.params.5:
 grads.network.params.6:
   device: cuda:0
   max: '2.984e-02'
-  mean: '-1.211e-09'
+  mean: '-5.588e-10'
   min: '-2.597e-02'
   shape:
   - 10
-  sum: '-1.211e-08'
+  sum: '-5.588e-09'
 grads.network.params.7:
   device: cuda:0
   max: '4.361e-02'
-  mean: '-3.26e-10'
+  mean: '-2.154e-10'
   min: '-4.662e-02'
   shape:
   - 256
   - 10
-  sum: '-8.345e-07'
+  sum: '-5.513e-07'
 outputs.logits:
   device: cuda:0
   max: '9.608e-01'

diff --git a/...ifier_test/test_backward_pass_is_reproducible/cifar10_jax_fcnet_jax_image_classifier.yaml b/...ifier_test/test_backward_pass_is_reproducible/cifar10_jax_fcnet_jax_image_classifier.yaml
@@ -37,20 +37,20 @@ grads.network.params.1:
 grads.network.params.2:
   device: cuda:0
   max: '6.868e-02'
-  mean: '-7.451e-10'
+  mean: '0.e+00'
   min: '-3.458e-02'
   shape:
   - 10
-  sum: '-7.451e-09'
+  sum: '0.e+00'
 grads.network.params.3:
   device: cuda:0
   max: '1.497e-01'
-  mean: '-4.191e-10'
+  mean: '-2.445e-10'
   min: '-1.415e-01'
   shape:
   - 256
   - 10
-  sum: '-1.073e-06'
+  sum: '-6.258e-07'
 outputs.logits:
   device: cuda:0
   max: '2.380e+00'

diff --git a/...r_test/test_backward_pass_is_reproducible/fashion_mnist_jax_cnn_jax_image_classifier.yaml b/...r_test/test_backward_pass_is_reproducible/fashion_mnist_jax_cnn_jax_image_classifier.yaml
@@ -83,12 +83,12 @@ grads.network.params.6:
 grads.network.params.7:
   device: cuda:0
   max: '1.382e-01'
-  mean: '-4.657e-10'
+  mean: '-1.775e-10'
   min: '-1.376e-01'
   shape:
   - 256
   - 10
-  sum: '-1.192e-06'
+  sum: '-4.545e-07'
 outputs.logits:
   device: cuda:0
   max: '1.032e+00'

diff --git a/...test/test_backward_pass_is_reproducible/fashion_mnist_jax_fcnet_jax_image_classifier.yaml b/...test/test_backward_pass_is_reproducible/fashion_mnist_jax_fcnet_jax_image_classifier.yaml
@@ -37,20 +37,20 @@ grads.network.params.1:
 grads.network.params.2:
   device: cuda:0
   max: '1.375e-01'
-  mean: '1.676e-09'
+  mean: '0.e+00'
   min: '-9.162e-02'
   shape:
   - 10
-  sum: '1.676e-08'
+  sum: '0.e+00'
 grads.network.params.3:
   device: cuda:0
   max: '3.990e-01'
-  mean: '2.328e-10'
+  mean: '-1.106e-10'
   min: '-2.054e-01'
   shape:
   - 256
   - 10
-  sum: '5.960e-07'
+  sum: '-2.831e-07'
 outputs.logits:
   device: cuda:0
   max: '2.656e+00'