diff --git a/applications/ColossalChat/coati/trainer/dpo.py b/applications/ColossalChat/coati/trainer/dpo.py index bd95e61b769c..eff5998fc3e7 100755 --- 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)