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

dont load best model

Browse source
This commit is contained in:
51616 2025-01-05 15:48:54 +00:00
parent fc30900703
commit 46c9ddb891
1 changed files with 12 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

24
hyperlora/configs.py
View file

@ -133,18 +133,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."},
)
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."},
)
# metric_for_best_model: str = field(
# default="val_loss",
# 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."},
# )
save_total_limit: int = field(
default=1,
metadata={"help": "Total number of checkpoints to save."},