yhavinga commited on
Commit
233fa8e
1 Parent(s): 617829a

Update run script parameters

Browse files
Files changed (1) hide show
  1. run_gpt.sh +1 -1
run_gpt.sh CHANGED
@@ -23,7 +23,7 @@ python run_clm_flax.py \
23
  --block_size="512" \
24
  --per_device_train_batch_size="16" \
25
  --per_device_eval_batch_size="16" \
26
- --learning_rate="8e-4" --warmup_steps="5000" \
27
  --adam_beta1="0.9" --adam_beta2="0.98" --weight_decay="0.01" \
28
  --overwrite_output_dir \
29
  --num_train_epochs="4" \
 
23
  --block_size="512" \
24
  --per_device_train_batch_size="16" \
25
  --per_device_eval_batch_size="16" \
26
+ --learning_rate="0.0024" --warmup_steps="5000" \
27
  --adam_beta1="0.9" --adam_beta2="0.98" --weight_decay="0.01" \
28
  --overwrite_output_dir \
29
  --num_train_epochs="4" \