From 1f3703ae96303496031585f87c0da8e73e7b81dc Mon Sep 17 00:00:00 2001 From: wql Date: Thu, 26 Sep 2024 15:02:21 +0800 Subject: [PATCH] chore: add print --- src/llamafactory/train/sft/workflow.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/src/llamafactory/train/sft/workflow.py b/src/llamafactory/train/sft/workflow.py index f02e5c86..502ac2f7 100644 --- a/src/llamafactory/train/sft/workflow.py +++ b/src/llamafactory/train/sft/workflow.py @@ -114,8 +114,15 @@ def run_sft( # Predict if training_args.do_predict: predict_results = trainer.predict(dataset_module["eval_dataset"], metric_key_prefix="predict", **gen_kwargs) + print("********************************************************") + print("********************************************************") + print("********************************************************") + print("********************************************************") + print("********************************************************") + print(predict_results.metrics) + print("********************************************************") time.sleep(100) if training_args.predict_with_generate: # predict_loss will be wrong if predict_with_generate is enabled predict_results.metrics.pop("predict_loss", None)