32 lines
465 B
TOML
32 lines
465 B
TOML
[run]
|
|
name = "remote_mlp_tiny"
|
|
seed = 0
|
|
artifact_dir = "artifacts/current_run/training_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 = 100
|
|
log_interval = 25
|
|
|
|
[device]
|
|
type = "cuda"
|
|
allow_cpu_fallback = false
|
|
benchmark_kernels = true
|
|
|
|
[loss]
|
|
type = "normalized_mse"
|