33 lines
438 B
TOML
33 lines
438 B
TOML
|
[run]
|
||
|
|
name = "mlp_tiny"
|
||
|
|
seed = 0
|
||
|
|
artifact_dir = "artifacts/runs"
|
||
|
|
|
||
|
|
[data]
|
||
|
|
root = "data/processed/minimal"
|
||
|
|
train_cases = 4
|
||
|
|
val_cases = 1
|
||
|
|
test_cases = 1
|
||
|
|
points_per_case = 128
|
||
|
|
batch_size = 128
|
||
|
|
|
||
|
|
[model]
|
||
|
|
type = "mlp"
|
||
|
|
hidden_width = 128
|
||
|
|
depth = 4
|
||
|
|
activation = "gelu"
|
||
|
|
|
||
|
|
[optim]
|
||
|
|
lr = 0.001
|
||
|
|
weight_decay = 0.0
|
||
|
|
steps = 500
|
||
|
|
log_interval = 100
|
||
|
|
|
||
|
|
[device]
|
||
|
|
type = "cuda"
|
||
|
|
allow_cpu_fallback = false
|
||
|
|
benchmark_kernels = true
|
||
|
|
|
||
|
|
[loss]
|
||
|
|
type = "normalized_mse"
|