fix

huggingface · Jun 18, 2024 · 7107831 · 7107831
1 parent 5ee6b01
commit 7107831
Show file tree

Hide file tree

Showing 13 changed files with 14 additions and 14 deletions.
diff --git a/docs/source/image_classification_params.mdx b/docs/source/image_classification_params.mdx
@@ -34,7 +34,7 @@
 --logging-steps LOGGING_STEPS
                     Determine how often to log training progress. Set this to the number of steps between each log output. -1 determines logging
                     steps automatically. Default is -1.
---evaluation-strategy {steps,epoch,no}
+--eval-strategy {steps,epoch,no}
                     Specify how often to evaluate the model performance. Options include 'no', 'steps', 'epoch'. 'epoch' evaluates at the end of
                     each training epoch by default.
 --save-total-limit SAVE_TOTAL_LIMIT

diff --git a/docs/source/llm_finetuning_params.mdx b/docs/source/llm_finetuning_params.mdx
@@ -130,7 +130,7 @@ The length parameters used for different trainers can be different. Some require
                     Set the dropout rate within the LoRA layers to help prevent overfitting during adaptation. Default is 0.05.
 --logging_steps LOGGING_STEPS, --logging-steps LOGGING_STEPS
                     Determine how often to log training progress in terms of steps. Setting it to '-1' determines logging steps automatically.
---eval_strategy {epoch,steps,no}, --evaluation-strategy {epoch,steps,no}
+--eval_strategy {epoch,steps,no}, --eval-strategy {epoch,steps,no}
                     Choose how frequently to evaluate the model's performance, with 'epoch' as the default, meaning at the end of each training
                     epoch
 --save_total_limit SAVE_TOTAL_LIMIT, --save-total-limit SAVE_TOTAL_LIMIT

diff --git a/docs/source/object_detection_params.mdx b/docs/source/object_detection_params.mdx
@@ -37,7 +37,7 @@
 --logging-steps LOGGING_STEPS
                     Determine how often to log training progress. Set this to the number of steps between each log output. -1 determines logging
                     steps automatically. Default is -1.
---evaluation-strategy {steps,epoch,no}
+--eval-strategy {steps,epoch,no}
                     Specify how often to evaluate the model performance. Options include 'no', 'steps', 'epoch'. 'epoch' evaluates at the end of
                     each training epoch by default.
 --save-total-limit SAVE_TOTAL_LIMIT

diff --git a/docs/source/seq2seq_params.mdx b/docs/source/seq2seq_params.mdx
@@ -42,7 +42,7 @@
 --logging-steps LOGGING_STEPS
                     Determine how often to log training progress. Set this to the number of steps between each log output. -1 determines logging
                     steps automatically. Default is -1.
---evaluation-strategy eval_strategy
+--eval-strategy eval_strategy
                     Specify how often to evaluate the model performance. Options include 'no', 'steps', 'epoch'. 'epoch' evaluates at the end of
                     each training epoch by default.
 --save-total-limit SAVE_TOTAL_LIMIT

diff --git a/docs/source/text_classification_params.mdx b/docs/source/text_classification_params.mdx
@@ -39,7 +39,7 @@
 --logging-steps LOGGING_STEPS
                     Determine how often to log training progress. Set this to the number of steps between each log output. -1 determines logging
                     steps automatically. Default is -1.
---evaluation-strategy {steps,epoch,no}
+--eval-strategy {steps,epoch,no}
                     Specify how often to evaluate the model performance. Options include 'no', 'steps', 'epoch'. 'epoch' evaluates at the end of
                     each training epoch by default.
 --save-total-limit SAVE_TOTAL_LIMIT

diff --git a/docs/source/text_regression.mdx b/docs/source/text_regression.mdx
@@ -68,7 +68,7 @@ usage: autotrain <command> [<args>] text-regression [-h] [--train] [--deploy] [-
                                                         [--log {none,wandb,tensorboard}] [--text-column TEXT_COLUMN] [--target-column TARGET_COLUMN]
                                                         [--max-seq-length MAX_SEQ_LENGTH] [--warmup-ratio WARMUP_RATIO] [--optimizer OPTIMIZER]
                                                         [--scheduler SCHEDULER] [--weight-decay WEIGHT_DECAY] [--max-grad-norm MAX_GRAD_NORM]
-                                                        [--logging-steps LOGGING_STEPS] [--evaluation-strategy {steps,epoch,no}]
+                                                        [--logging-steps LOGGING_STEPS] [--eval-strategy {steps,epoch,no}]
                                                         [--save-total-limit SAVE_TOTAL_LIMIT]
                                                         [--auto-find-batch-size] [--mixed-precision {fp16,bf16,None}]
 
@@ -132,7 +132,7 @@ options:
   --logging-steps LOGGING_STEPS
                         Determine how often to log training progress. Set this to the number of steps between each log output. -1 determines logging
                         steps automatically. Default is -1.
-  --evaluation-strategy {steps,epoch,no}
+  --eval-strategy {steps,epoch,no}
                         Specify how often to evaluate the model performance. Options include 'no', 'steps', 'epoch'. 'epoch' evaluates at the end of
                         each training epoch by default.
   --save-total-limit SAVE_TOTAL_LIMIT

diff --git a/docs/source/token_classification_params.mdx b/docs/source/token_classification_params.mdx
@@ -37,7 +37,7 @@
 --logging-steps LOGGING_STEPS
                     Determine how often to log training progress. Set this to the number of steps between each log output. -1 determines logging
                     steps automatically. Default is -1.
---evaluation-strategy {steps,epoch,no}
+--eval-strategy {steps,epoch,no}
                     Specify how often to evaluate the model performance. Options include 'no', 'steps', 'epoch'. 'epoch' evaluates at the end of
                     each training epoch by default.
 --save-total-limit SAVE_TOTAL_LIMIT

diff --git a/src/autotrain/cli/run_image_classification.py b/src/autotrain/cli/run_image_classification.py
@@ -73,7 +73,7 @@ def register_subcommand(parser: ArgumentParser):
                 "default": -1,
             },
             {
-                "arg": "--evaluation-strategy",
+                "arg": "--eval-strategy",
                 "help": "Specify how often to evaluate the model performance. Options include 'no', 'steps', 'epoch'. 'epoch' evaluates at the end of each training epoch by default.",
                 "required": False,
                 "type": str,

diff --git a/src/autotrain/cli/run_llm.py b/src/autotrain/cli/run_llm.py
@@ -145,7 +145,7 @@ def register_subcommand(parser: ArgumentParser):
                 "required": False,
                 "type": str,
                 "default": "epoch",
-                "alias": ["--evaluation-strategy"],
+                "alias": ["--eval-strategy"],
                 "choices": ["epoch", "steps", "no"],
             },
             {

diff --git a/src/autotrain/cli/run_seq2seq.py b/src/autotrain/cli/run_seq2seq.py
@@ -87,7 +87,7 @@ def register_subcommand(parser: ArgumentParser):
                 "default": -1,
             },
             {
-                "arg": "--evaluation-strategy",
+                "arg": "--eval-strategy",
                 "help": "Specify how often to evaluate the model performance. Options include 'no', 'steps', 'epoch'. 'epoch' evaluates at the end of each training epoch by default.",
                 "required": False,
                 "type": str,

diff --git a/src/autotrain/cli/run_text_classification.py b/src/autotrain/cli/run_text_classification.py
@@ -80,7 +80,7 @@ def register_subcommand(parser: ArgumentParser):
                 "default": -1,
             },
             {
-                "arg": "--evaluation-strategy",
+                "arg": "--eval-strategy",
                 "help": "Specify how often to evaluate the model performance. Options include 'no', 'steps', 'epoch'. 'epoch' evaluates at the end of each training epoch by default.",
                 "required": False,
                 "type": str,

diff --git a/src/autotrain/cli/run_text_regression.py b/src/autotrain/cli/run_text_regression.py
@@ -80,7 +80,7 @@ def register_subcommand(parser: ArgumentParser):
                 "default": -1,
             },
             {
-                "arg": "--evaluation-strategy",
+                "arg": "--eval-strategy",
                 "help": "Specify how often to evaluate the model performance. Options include 'no', 'steps', 'epoch'. 'epoch' evaluates at the end of each training epoch by default.",
                 "required": False,
                 "type": str,

diff --git a/src/autotrain/cli/run_token_classification.py b/src/autotrain/cli/run_token_classification.py
@@ -80,7 +80,7 @@ def register_subcommand(parser: ArgumentParser):
                 "default": -1,
             },
             {
-                "arg": "--evaluation-strategy",
+                "arg": "--eval-strategy",
                 "help": "Specify how often to evaluate the model performance. Options include 'no', 'steps', 'epoch'. 'epoch' evaluates at the end of each training epoch by default.",
                 "required": False,
                 "type": str,