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,28 @@
task:
name: mnist
optimizer:
- name: adamw_baseline
learning_rate: [1.e-1, 1.e-2, 1.e-3]
# learning_rate: [1.e-1, 3.16e-2, 1.e-2, 3.16e-3, 1.e-3] # finer grid
weight_decay: [1.e-0, 1.e-1, 1.e-2, 1.e-3]
- name: adamcpr_fast
learning_rate: [1.e-1, 1.e-2, 1.e-3]
# learning_rate: [1.e-1, 3.16e-2, 1.e-2, 3.16e-3, 1.e-3] # finer grid
kappa_init_param: [0.5, 1, 2, 4, 8, 16, 32]
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.8 # increase this for slower machine
sbatch_args:
partition: single # adapt to your cluster
evaluation:
output_types: [pdf]
plot:
x_axis:
- optimizer.kappa_init_param
- optimizer.weight_decay