Spaces:
Running
Running
feat: update test script
Browse files- seq2seq/do_run.sh +2 -2
seq2seq/do_run.sh
CHANGED
@@ -6,11 +6,11 @@ python run_seq2seq_flax.py \
|
|
6 |
--per_device_train_batch_size 56 \
|
7 |
--per_device_eval_batch_size 56 \
|
8 |
--preprocessing_num_workers 80 \
|
9 |
-
--warmup_steps
|
10 |
--gradient_accumulation_steps 8 \
|
11 |
--do_train \
|
12 |
--do_eval \
|
13 |
--adafactor \
|
14 |
--num_train_epochs 1 \
|
15 |
-
--max_train_samples
|
16 |
--learning_rate 0.003
|
|
|
6 |
--per_device_train_batch_size 56 \
|
7 |
--per_device_eval_batch_size 56 \
|
8 |
--preprocessing_num_workers 80 \
|
9 |
+
--warmup_steps 125 \
|
10 |
--gradient_accumulation_steps 8 \
|
11 |
--do_train \
|
12 |
--do_eval \
|
13 |
--adafactor \
|
14 |
--num_train_epochs 1 \
|
15 |
+
--max_train_samples 20000 \
|
16 |
--learning_rate 0.003
|