fix args
This commit is contained in:
1 parent
17c64a0579
commit
d92f112951
4 files changed
+8
-8
No files matched your search
@@ -60,9 +60,9 @@ def run_dpo(
|
|||||||
plot_loss(training_args.output_dir, keys=["loss", "eval_loss"])
|
plot_loss(training_args.output_dir, keys=["loss", "eval_loss"])
|
||||||
|
|
||||||
if training_args.push_to_hub:
|
if training_args.push_to_hub:
|
||||||
trainer.push_to_hub(**generate_model_card())
|
trainer.push_to_hub(**generate_model_card(model_args, data_args, finetuning_args))
|
||||||
else:
|
else:
|
||||||
trainer.create_model_card(**generate_model_card())
|
trainer.create_model_card(**generate_model_card(model_args, data_args, finetuning_args))
|
||||||
|
|
||||||
# Evaluation
|
# Evaluation
|
||||||
if training_args.do_eval:
|
if training_args.do_eval:
|
||||||
|
|||||||
@@ -46,9 +46,9 @@ def run_pt(
|
|||||||
plot_loss(training_args.output_dir, keys=["loss", "eval_loss"])
|
plot_loss(training_args.output_dir, keys=["loss", "eval_loss"])
|
||||||
|
|
||||||
if training_args.push_to_hub:
|
if training_args.push_to_hub:
|
||||||
trainer.push_to_hub(**generate_model_card())
|
trainer.push_to_hub(**generate_model_card(model_args, data_args, finetuning_args))
|
||||||
else:
|
else:
|
||||||
trainer.create_model_card(**generate_model_card())
|
trainer.create_model_card(**generate_model_card(model_args, data_args, finetuning_args))
|
||||||
|
|
||||||
# Evaluation
|
# Evaluation
|
||||||
if training_args.do_eval:
|
if training_args.do_eval:
|
||||||
|
|||||||
@@ -54,9 +54,9 @@ def run_rm(
|
|||||||
plot_loss(training_args.output_dir, keys=["loss", "eval_loss"])
|
plot_loss(training_args.output_dir, keys=["loss", "eval_loss"])
|
||||||
|
|
||||||
if training_args.push_to_hub:
|
if training_args.push_to_hub:
|
||||||
trainer.push_to_hub(**generate_model_card())
|
trainer.push_to_hub(**generate_model_card(model_args, data_args, finetuning_args))
|
||||||
else:
|
else:
|
||||||
trainer.create_model_card(**generate_model_card())
|
trainer.create_model_card(**generate_model_card(model_args, data_args, finetuning_args))
|
||||||
|
|
||||||
# Evaluation
|
# Evaluation
|
||||||
if training_args.do_eval:
|
if training_args.do_eval:
|
||||||
|
|||||||
@@ -73,9 +73,9 @@ def run_sft(
|
|||||||
plot_loss(training_args.output_dir, keys=["loss", "eval_loss"])
|
plot_loss(training_args.output_dir, keys=["loss", "eval_loss"])
|
||||||
|
|
||||||
if training_args.push_to_hub:
|
if training_args.push_to_hub:
|
||||||
trainer.push_to_hub(**generate_model_card())
|
trainer.push_to_hub(**generate_model_card(model_args, data_args, finetuning_args))
|
||||||
else:
|
else:
|
||||||
trainer.create_model_card(**generate_model_card())
|
trainer.create_model_card(**generate_model_card(model_args, data_args, finetuning_args))
|
||||||
|
|
||||||
# Evaluation
|
# Evaluation
|
||||||
if training_args.do_eval:
|
if training_args.do_eval:
|
||||||
|
|||||||
Reference in new issue
Block a user