Update canyon_med.py

Load NN weights as float 64 instead of 128
euroargodev · Oct 11, 2024 · 3e7d0d2 · 3e7d0d2
1 parent 00ba5b1
commit 3e7d0d2
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/argopy/extensions/canyon_med.py b/argopy/extensions/canyon_med.py
@@ -150,9 +150,9 @@ def load_weights(self, param, subset, i):
 
         # Using float128 arrays avoid the error or warning "overflow encountered in exp" raised by the
         # activation function
-        b1 = np.array(b1, dtype=np.float128)
-        b2 = np.array(b2, dtype=np.float128)
-        b3 = np.array(b3, dtype=np.float128)
+        b1 = np.array(b1, dtype=np.float64)
+        b2 = np.array(b2, dtype=np.float64)
+        b3 = np.array(b3, dtype=np.float64)
 
         return b1, b2, b3, IW, LW1, LW2