From 380408eae83982b5dadbef44146a18dc37780c1e Mon Sep 17 00:00:00 2001 From: Tong Li Date: Tue, 10 Sep 2024 10:10:10 +0000 Subject: [PATCH] update msg --- applications/ColossalChat/coati/trainer/dpo.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)