linting, moved env, updated contrib credit

This commit is contained in:
Shannon Sands 2025-05-26 14:35:16 +10:00
parent 81d1ebeaef
commit bf12e7df15
83 changed files with 1560 additions and 640 deletions

View file

@ -0,0 +1,27 @@
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