apunkt/doc-to-lora
1
0
Fork 0
mirror of https://github.com/SakanaAI/doc-to-lora.git synced 2026-07-23 17:01:04 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix metric_for_best_model default

Browse source
This commit is contained in:
51616 2025-06-23 13:31:38 +09:00
parent 00065320d8
commit 88d5c375ed
1 changed files with 8 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

16
src/ctx_to_lora/configs.py
View file

@ -201,18 +201,18 @@ class TrainingArguments(TrainingArguments):
default=10_000,
metadata={"help": "Evaluation steps."},
)
# metric_for_best_model: str = field(
# default="val_loss",
# metadata={"help": "Metric for best model."},
# )
metric_for_best_model: str = field(
default=None,
metadata={"help": "Metric for best model."},
)
# greater_is_better: bool = field(
# default=False,
# metadata={"help": "Whether the metric is better when it is greater."},
# )
# load_best_model_at_end: bool = field(
# default=False,
# metadata={"help": "Whether to load the best model at the end of training."},
# )
load_best_model_at_end: bool = field(
default=False,
metadata={"help": "Whether to load the best model at the end of training."},
)
save_total_limit: int = field(
default=2,
metadata={"help": "Total number of checkpoints to save."},