fix example params

This commit is contained in:
hiyouga
2024-03-08 20:41:43 +08:00
parent 4a2cc60b94
commit 8a45213440
4 changed files with 5 additions and 5 deletions

View File

@@ -15,7 +15,7 @@ CUDA_VISIBLE_DEVICES=0 python ../../../src/train_bash.py \
--preprocessing_num_workers 16 \
--per_device_train_batch_size 1 \
--per_device_eval_batch_size 1 \
--gradient_accumulation_steps 8 \
--gradient_accumulation_steps 2 \
--lr_scheduler_type cosine \
--logging_steps 10 \
--warmup_steps 20 \