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: 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
|
||||
|
|
@ -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
|
||||
|
|
@ -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
|
||||
Loading…
Add table
Add a link
Reference in a new issue