fix(tests): fix ci lint and test err

InternLM · Sep 6, 2024 · 2758409 · 2758409
1 parent 6695bba
commit 2758409
Showing 1 changed file with 10 additions and 10 deletions.
diff --git a/tests/test_training/test_loss.py b/tests/test_training/test_loss.py
@@ -465,16 +465,16 @@ def test_training_with_isp():
     global CONFIG_FILE_PATH, BASELINE_LOSS_LIST
     CONFIG_FILE_PATH = "./configs/7B_isp_sft.py"
     BASELINE_LOSS_LIST = [
-        11.579612731933594,
-        8.526281356811523,
-        7.188107490539551,
-        6.010709762573242,
-        5.51548957824707,
-        5.881415367126465,
-        4.9155402183532715,
-        5.00706672668457,
-        4.36771297454834,
-        4.7433953285217285,
+        11.595988273620605,
+        7.988386154174805,
+        6.821506500244141,
+        6.2768449783325195,
+        5.478013515472412,
+        5.4622697830200195,
+        5.162247180938721,
+        4.854615211486816,
+        4.744818210601807,
+        4.75523567199707,
     ]
 
     # model training