p04798526
/
LLaMA-Factory-310P3
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

chore: add print

This commit is contained in:
wql 2024-09-26 15:02:21 +08:00
parent 1ac91733e4
commit 1f3703ae96
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/llamafactory/train/sft/workflow.py
View File

@ -114,8 +114,15 @@ def run_sft(
# Predict # Predict
if training_args.do_predict: if training_args.do_predict:
predict_results = trainer.predict(dataset_module["eval_dataset"], metric_key_prefix="predict", **gen_kwargs) predict_results = trainer.predict(dataset_module["eval_dataset"], metric_key_prefix="predict", **gen_kwargs)
print("********************************************************")
print("********************************************************")
print("********************************************************")
print("********************************************************")
print("********************************************************")
print(predict_results.metrics) print(predict_results.metrics)
print("********************************************************")
time.sleep(100) time.sleep(100)
if training_args.predict_with_generate: # predict_loss will be wrong if predict_with_generate is enabled if training_args.predict_with_generate: # predict_loss will be wrong if predict_with_generate is enabled
predict_results.metrics.pop("predict_loss", None) predict_results.metrics.pop("predict_loss", None)