boris commited on
Commit
3cccb01
1 Parent(s): 533b494

feat: update test script

Browse files
Files changed (1) hide show
  1. 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 1000 \
10
  --gradient_accumulation_steps 8 \
11
  --do_train \
12
  --do_eval \
13
  --adafactor \
14
  --num_train_epochs 1 \
15
- --max_train_samples 2000 \
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