fix: trainer post-pretrain log route-prefix (#266)

baidubce · Feb 6, 2024 · bb829fd · bb829fd
1 parent 8d75c76
commit bb829fd
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/python/qianfan/trainer/actions.py b/python/qianfan/trainer/actions.py
@@ -534,11 +534,16 @@ def _wait_model_trained(self, **kwargs: Dict) -> None:
             elif task_status == console_consts.TrainStatus.Running:
                 job_progress_str = task_status_result.get("runProgress")
                 job_progress = int(job_progress_str[:-1])
+                log_prefix = (
+                    "sft"
+                    if self.train_mode == console_consts.TrainMode.SFT
+                    else "postPretrain"
+                )
                 log_info(
                     "[train_action] training ..."
                     f" job_name:{self.job_name} current status: {task_status},"
                     f" {job_progress}% check train task log in"
-                    f" https://console.bce.baidu.com/qianfan/train/sft/{self.job_id}/{self.task_id}/detail/traininglog"
+                    f" https://console.bce.baidu.com/qianfan/train/{log_prefix}/{self.job_id}/{self.task_id}/detail/traininglog"
                 )
                 if job_progress >= 50:
                     log_info(