Fix(QA): fix monthly test (#202)

InternLM · Apr 16, 2024 · b024cf4 · b024cf4
1 parent 7d9428b
commit b024cf4
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/tests/common_fixture.py b/tests/common_fixture.py
@@ -36,6 +36,7 @@
             diag_outlier_ratio=1.1,
             train_folder=None,
             valid_folder=None,
+            num_worker=0,
         ),
         model=dict(
             checkpoint=False,

diff --git a/tests/test_training/test_norm_weight.py b/tests/test_training/test_norm_weight.py
@@ -106,6 +106,8 @@ def train_check_norm_weight(args):
     train_iter = iter(train_dl)
 
     for batch_count in range(total_steps):
+        if gpc.is_rank_for_log() and batch_count % 100 == 0:
+            print(f"batch_count: {batch_count}", flush=True)
         if batch_count % 100 == 0:
             internlm_accelerator.empty_cache()
             gc.collect()
@@ -180,6 +182,7 @@ def test_check_norm_msp():
         pool.join()
 
     check_result(result)
+    print("msp check pass", flush=True)
 
 
 @pytest.mark.check_norm_fsp
@@ -195,6 +198,7 @@ def test_check_norm_fsp():
         pool.join()
 
     check_result(result)
+    print("fsp check pass", flush=True)
 
 
 @pytest.mark.check_norm_isp
@@ -210,3 +214,4 @@ def test_check_norm_isp():
         pool.join()
 
     check_result(result)
+    print("isp check pass", flush=True)