Merge remote-tracking branch 'upstream/main'

red-hat-data-services · May 24, 2024 · 019ca6a · 019ca6a
2 parents e1e8dd9 + 697b573
commit 019ca6a
Show file tree

Hide file tree

Showing 4 changed files with 134 additions and 263 deletions.
diff --git a/README.md b/README.md
@@ -80,7 +80,7 @@ python tuning/sft_trainer.py  \
 --per_device_train_batch_size 4  \
 --per_device_eval_batch_size 4  \
 --gradient_accumulation_steps 4  \
---evaluation_strategy "no"  \
+--eval_strategy "no"  \
 --save_strategy "epoch"  \
 --learning_rate 1e-5  \
 --weight_decay 0.  \
@@ -125,7 +125,7 @@ tuning/sft_trainer.py \
 --per_device_train_batch_size 4 \
 --per_device_eval_batch_size 4 \
 --gradient_accumulation_steps 4 \
---evaluation_strategy "no" \
+--eval_strategy "no" \
 --save_strategy "epoch" \
 --learning_rate 1e-5 \
 --weight_decay 0. \
@@ -279,7 +279,7 @@ tuning/sft_trainer.py  \
 --per_device_train_batch_size 1  \
 --per_device_eval_batch_size 1  \
 --gradient_accumulation_steps 1  \
---evaluation_strategy "no"  \
+--eval_strategy "no"  \
 --save_strategy "epoch"  \
 --learning_rate 1e-5  \
 --weight_decay 0.  \
@@ -313,7 +313,7 @@ tuning/sft_trainer.py  \
 --per_device_train_batch_size 1  \
 --per_device_eval_batch_size 1  \
 --gradient_accumulation_steps 1  \
---evaluation_strategy "no"  \
+--eval_strategy "no"  \
 --save_strategy "epoch"  \
 --learning_rate 1e-5  \
 --weight_decay 0.  \

diff --git a/examples/prompt_tuning_twitter_complaints/README.md b/examples/prompt_tuning_twitter_complaints/README.md
@@ -51,7 +51,7 @@ tuning/sft_trainer.py  \
 --per_device_train_batch_size 1  \
 --per_device_eval_batch_size 1  \
 --gradient_accumulation_steps 1  \
---evaluation_strategy "no"  \
+--eval_strategy "no"  \
 --save_strategy "epoch"  \
 --learning_rate 1e-5  \
 --weight_decay 0.  \

diff --git a/tests/helpers.py b/tests/helpers.py