mirror of
https://github.com/NousResearch/atropos.git
synced 2026-04-19 12:57:58 +00:00
27 lines
763 B
YAML
27 lines
763 B
YAML
task:
|
|
name: segmentation
|
|
optimizer:
|
|
- name: adamw_baseline
|
|
learning_rate: [3.16e-3, 1.e-3, 3.16e-4]
|
|
weight_decay: [1.e-1, 1.e-2, 1.e-3, 0]
|
|
- name: adamcpr_fast
|
|
learning_rate: [3.16e-3, 1.e-3, 3.16e-4]
|
|
kappa_init_param: [1, 4, 16, 64]
|
|
engine:
|
|
seed: [1, 2, 3]
|
|
# data_dir: ./data
|
|
# output_dir: ./experiments
|
|
plot: false
|
|
silent: true
|
|
sbatch_script_template: baselines/sbatch_template.sh # adapt the template to your needs
|
|
run_scheduler: slurm_array
|
|
sbatch_time_factor: 1.5 # increase this for slower machine
|
|
sbatch_args:
|
|
partition: single # adapt to your cluster
|
|
save_sbatch_scripts: slurm-scripts
|
|
evaluation:
|
|
output_types: [pdf]
|
|
plot:
|
|
x_axis:
|
|
- optimizer.kappa_init_param
|
|
- optimizer.weight_decay
|