frozenleaves/trl
1
0
Fork 0
mirror of https://github.com/huggingface/trl.git synced 2025-10-21 02:53:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

Standardize pushing to Hub in examples (#2126)

Browse Source
This commit is contained in:
Quentin Gallouédec
2024-09-26 10:00:51 +02:00
committed by GitHub
parent fb1b48fdbe
commit 32d9d34eb1
22 changed files with 88 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
examples/scripts/reward_modeling.py
View File

@ -126,5 +126,8 @@ if __name__ == "__main__":
metrics = trainer.evaluate()
trainer.log_metrics("eval", metrics)
trainer.save_metrics("eval", metrics)
# Save and push to hub
trainer.save_model(training_args.output_dir)
trainer.push_to_hub()
if training_args.push_to_hub:
trainer.push_to_hub()