mirror of
https://github.com/NousResearch/atropos.git
synced 2026-04-19 12:57:58 +00:00
linting, moved env, updated contrib credit
This commit is contained in:
parent
81d1ebeaef
commit
bf12e7df15
83 changed files with 1560 additions and 640 deletions
|
|
@ -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
|
||||
Loading…
Add table
Add a link
Reference in a new issue