Update regression files (agAIN!)

Signed-off-by: Fabrice Normandin <[email protected]>
mila-iqia · Nov 23, 2024 · 0391ca5 · 0391ca5
1 parent 61ecc0c
commit 0391ca5
Show file tree

Hide file tree

Showing 7 changed files with 866 additions and 866 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: '-3.725e-10'
+  mean: '-5.588e-10'
   min: '-2.597e-02'
   shape:
   - 10
-  sum: '-3.725e-09'
+  sum: '-5.588e-09'
 grads.network.params.7:
   device: cuda:0
   max: '4.361e-02'
-  mean: '-7.567e-11'
+  mean: '-2.154e-10'
   min: '-4.662e-02'
   shape:
   - 256
   - 10
-  sum: '-1.937e-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: '-5.937e-10'
+  mean: '-2.445e-10'
   min: '-1.415e-01'
   shape:
   - 256
   - 10
-  sum: '-1.52e-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
@@ -75,20 +75,20 @@ grads.network.params.5:
 grads.network.params.6:
   device: cuda:0
   max: '5.898e-02'
-  mean: '-2.235e-09'
+  mean: '-1.863e-09'
   min: '-7.022e-02'
   shape:
   - 10
-  sum: '-2.235e-08'
+  sum: '-1.863e-08'
 grads.network.params.7:
   device: cuda:0
   max: '1.382e-01'
-  mean: '-3.609e-10'
+  mean: '-1.775e-10'
   min: '-1.376e-01'
   shape:
   - 256
   - 10
-  sum: '-9.239e-07'
+  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
@@ -45,12 +45,12 @@ grads.network.params.2:
 grads.network.params.3:
   device: cuda:0
   max: '3.990e-01'
-  mean: '-2.910e-11'
+  mean: '-1.106e-10'
   min: '-2.054e-01'
   shape:
   - 256
   - 10
-  sum: '-7.451e-08'
+  sum: '-2.831e-07'
 outputs.logits:
   device: cuda:0
   max: '2.656e+00'

diff --git a/...lassifier_test/test_backward_pass_is_reproducible/mnist_jax_cnn_jax_image_classifier.yaml b/...lassifier_test/test_backward_pass_is_reproducible/mnist_jax_cnn_jax_image_classifier.yaml
@@ -21,40 +21,40 @@ grads.network.params.0:
   device: cuda:0
   max: '1.65e-02'
   mean: '2.109e-03'
-  min: '-8.631e-03'
+  min: '-8.628e-03'
   shape:
   - 32
-  sum: '6.747e-02'
+  sum: '6.748e-02'
 grads.network.params.1:
   device: cuda:0
-  max: '1.894e-02'
-  mean: '-1.554e-05'
-  min: '-1.628e-02'
+  max: '1.893e-02'
+  mean: '-1.55e-05'
+  min: '-1.627e-02'
   shape:
   - 3
   - 3
   - 1
   - 32
-  sum: '-4.475e-03'
+  sum: '-4.463e-03'
 grads.network.params.2:
   device: cuda:0
   max: '2.053e-02'
   mean: '1.196e-03'
   min: '-1.783e-02'
   shape:
   - 64
-  sum: '7.655e-02'
+  sum: '7.653e-02'
 grads.network.params.3:
   device: cuda:0
   max: '2.25e-02'
-  mean: '3.614e-04'
+  mean: '3.613e-04'
   min: '-2.352e-02'
   shape:
   - 3
   - 3
   - 32
   - 64
-  sum: '6.662e+00'
+  sum: '6.659e+00'
 grads.network.params.4:
   device: cuda:0
   max: '2.231e-02'
@@ -75,20 +75,20 @@ grads.network.params.5:
 grads.network.params.6:
   device: cuda:0
   max: '6.484e-02'
-  mean: '-2.980e-09'
+  mean: '-1.490e-09'
   min: '-8.046e-02'
   shape:
   - 10
-  sum: '-2.980e-08'
+  sum: '-1.490e-08'
 grads.network.params.7:
   device: cuda:0
   max: '7.496e-02'
-  mean: '-3.754e-10'
+  mean: '-3.361e-10'
   min: '-8.565e-02'
   shape:
   - 256
   - 10
-  sum: '-9.611e-07'
+  sum: '-8.605e-07'
 outputs.logits:
   device: cuda:0
   max: '8.092e-01'

diff --git a/...ssifier_test/test_backward_pass_is_reproducible/mnist_jax_fcnet_jax_image_classifier.yaml b/...ssifier_test/test_backward_pass_is_reproducible/mnist_jax_fcnet_jax_image_classifier.yaml
@@ -37,20 +37,20 @@ grads.network.params.1:
 grads.network.params.2:
   device: cuda:0
   max: '4.549e-02'
-  mean: '-3.725e-10'
+  mean: '0.e+00'
   min: '-7.537e-02'
   shape:
   - 10
-  sum: '-3.725e-09'
+  sum: '0.e+00'
 grads.network.params.3:
   device: cuda:0
   max: '7.07e-02'
-  mean: '-3.929e-10'
+  mean: '-5.821e-11'
   min: '-1.064e-01'
   shape:
   - 256
   - 10
-  sum: '-1.006e-06'
+  sum: '-1.490e-07'
 outputs.logits:
   device: cuda:0
   max: '1.85e+00'