tiny fix

Former-commit-id: eae79707d31fd8be2cf4bee4d610557bbd49f6e7
2023-05-29 09:42:29 +08:00
parent 83fc73c580
commit 35d04a2c05
5 changed files with 12 additions and 12 deletions
--- a/src/train_rm.py
+++ b/src/train_rm.py
@@ -55,7 +55,7 @@ def main():
        trainer.save_metrics("train", train_result.metrics)
        trainer.save_state()
        trainer.save_model()
-        if trainer.is_world_process_zero() and finetuning_args.plot_loss:
+        if trainer.is_world_process_zero() and model_args.plot_loss:
            plot_loss(training_args, keys=["loss", "eval_loss"])

    # Evaluation