Update train_CI.py

InternLM · Sep 4, 2024 · a76e14e · a76e14e
1 parent ef2bec7
commit a76e14e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_training/train_CI.py b/tests/test_training/train_CI.py
@@ -305,7 +305,7 @@ def main(args):
                 train_state=train_state,
                 optimizer=optimizer,
                 beta2_scheduler=beta2_scheduler,
-                trainer=trainer,
+                engine=trainer.engine,
                 very_begining_time=very_begining_time,
                 start_time=start_time,
                 loss=loss,