kyy/llm_trainer
1
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix #1567

Browse Source
This commit is contained in:
hiyouga committed 2023-11-20 18:46:36 +08:00
1 parent 00baaa990e
commit 99a3f06377
5 files changed
+34 -36
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

src/llmtuner/train/rm/workflow.py
+1 -1
View File
@@ -47,7 +47,7 @@ def run_rm(
# Training
if training_args.do_train:
train_result = trainer.train()
train_result = trainer.train(resume_from_checkpoint=training_args.resume_from_checkpoint)
trainer.save_model()
trainer.log_metrics("train", train_result.metrics)
trainer.save_metrics("train", train_result.metrics)