From 46c9ddb891843266478c3f459e8c15a3fe7686a7 Mon Sep 17 00:00:00 2001 From: 51616 Date: Sun, 5 Jan 2025 15:48:54 +0000 Subject: [PATCH] dont load best model --- hyperlora/configs.py | 24 ++++++++++++------------ 1 file changed, 12 insertions(+), 12 deletions(-) diff --git a/hyperlora/configs.py b/hyperlora/configs.py index 1ff4554..6c4cf2f 100644 --- a/hyperlora/configs.py +++ b/hyperlora/configs.py @@ -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."},