update msg

TongLi3701 · Sep 10, 2024 · 380408e · 380408e
1 parent a732586
commit 380408e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/applications/ColossalChat/coati/trainer/dpo.py b/applications/ColossalChat/coati/trainer/dpo.py
@@ -357,7 +357,7 @@ def _eval(self, epoch: int):
                 )
                 step_bar.update()
 
-        msg = "Evaluation Result:\n"
+        msg = "\nEvaluation Result:\n"
         for tag in ["loss", "chosen_rewards", "rejected_rewards", "accuracy", "margin"]:
             msg = msg + f"{tag}: {self.accumulative_meter.get(tag)}\n"
         self.coordinator.print_on_master(msg)