update rtols

NVIDIA · Oct 10, 2023 · 69b5746 · 69b5746
1 parent 9987b10
commit 69b5746
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/.github/workflows/baselines/test_t5x_mgmn_metrics.py b/.github/workflows/baselines/test_t5x_mgmn_metrics.py
@@ -8,17 +8,17 @@
 
 LOSS_RTOL = {
     '1G1N': 0.01,
-    '1G2N': 0.015,
-    '1P1G': 0.015,
-    '1P2G': 0.015,
-    '1P4G': 0.015,
-    '1P8G': 0.015,
+    '1G2N': 0.03,
+    '1P1G': 0.03,
+    '1P2G': 0.03,
+    '1P4G': 0.03,
+    '1P8G': 0.035,
     '2G1N': 0.015,
     '2G2N': 0.015,
     '4G1N': 0.015,
     '4G2N': 0.02,
     '8G1N': 0.02,
-    '8G2N': 0.02
+    '8G2N': 0.035
 }
 STEP_TIME_MULT = {
     "1G1N": 0.95,