Spaces:
Running
Running
File size: 918 Bytes
dad6d93 3073ff4 dbe8c41 dbbd01a dad6d93 dbe8c41 dad6d93 8f058ae dad6d93 dbe8c41 dad6d93 2f69241 dad6d93 2f69241 dad6d93 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 |
program: run_seq2seq_flax.py
entity: wandb
project: hf-flax-dalle-mini
method: random
metric:
name: eval/loss
goal: minimize
parameters:
learning_rate:
distribution: log_uniform
# from exp(min) to exp(max), ie 5e-5 to 5e-3 on log scale
min: -9.9
max: -5.3
gradient_accumulation_steps:
value: 8
warmup_steps:
# in term of optimization steps so multiplied by gradient accumulation
value: 125
command:
- python3
- ${program}
- "--train_file"
- "/data/CC12M/encoded-small-train.tsv"
- "--validation_file"
- "/data/CC12M/encoded-small-valid.tsv"
- "--output_dir"
- "./output_sweep"
- "--overwrite_output_dir"
- "--adafactor"
- "--num_train_epochs"
- 1
- "--max_train_samples"
- 1500000
- "--per_device_train_batch_size"
- 56
- "--per_device_eval_batch_size"
- 56
- "--preprocessing_num_workers"
- 80
- "--do_train"
- "--do_eval"
- ${args}
|