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: mnist
optimizer:
name: adamw_baseline
learning_rate: [1.0e-2, 1.0e-3]
weight_decay: [0.1, 0.01]
engine:
data_dir: ./examples/data
output_dir: ./examples/plotting/outputs
seed: [1, 2]
evaluation:
output_dir: ./examples/plotting
experiment_name: 1_mnist-adamw
checkpoints:
- last
output_types:
- png
plot:
std: True
x_axis:
- optimizer.weight_decay
y_axis:
- optimizer.learning_rate
plotstyle:
tight_layout: False # the title is a little bit squeezed with this
text:
usetex: True

View file

@ -0,0 +1,30 @@
task:
name: mnist
optimizer:
- name: sgd_baseline
learning_rate: [0.01, 0.001]
weight_decay: [0.1, 0.01]
- name: adamw_baseline
learning_rate: [0.01, 0.001]
weight_decay: [0.1, 0.01]
engine:
data_dir: ./examples/data
output_dir: ./examples/plotting/outputs
seed: [1, 2]
evaluation:
output_dir: ./examples/plotting
experiment_name: 2_adamw-vs-sgd
checkpoints:
- last
output_types:
- png
plot:
std: True
x_axis:
- optimizer.weight_decay
y_axis:
- optimizer.learning_rate
plotstyle:
tight_layout: False # the title is a little bit squeezed with this
text:
usetex: True

View file

@ -0,0 +1,32 @@
task:
- mnist
- tabular
optimizer:
- name: sgd_baseline
learning_rate: [0.01, 0.001]
weight_decay: [0.1, 0.01]
- name: adamw_baseline
learning_rate: [0.01, 0.001]
weight_decay: [0.1, 0.01]
engine:
data_dir: ./examples/data
output_dir: ./examples/plotting/outputs
seed: [1, 2]
evaluation:
output_dir: ./examples/plotting
experiment_name: 3_mnist-and-tabular_adamw-vs-sgd
checkpoints:
- last
output_types:
- png
split_groups: ["task.name"]
plot:
std: True
x_axis:
- optimizer.weight_decay
y_axis:
- optimizer.learning_rate
plotstyle:
tight_layout: False # the title is a little bit squeezed with this
text:
usetex: True