| { |
| "bundle": { |
| "bundle_subdir": "bundle" |
| }, |
| "hub": { |
| "repo_id": null, |
| "revision": null |
| }, |
| "io": { |
| "x_dim": 3, |
| "x_names": null, |
| "y_dim": 2, |
| "y_names": null |
| }, |
| "model": { |
| "name": "mlp", |
| "params": { |
| "activation": "gelu", |
| "dtype": "float32", |
| "hidden_sizes": [ |
| 32, |
| 32 |
| ] |
| } |
| }, |
| "optim": { |
| "b1": 0.9, |
| "b2": 0.999, |
| "decay_steps": 0, |
| "eps": 1e-08, |
| "lr": 0.0003, |
| "name": "adamw", |
| "precondition_1d": false, |
| "precondition_frequency": 10, |
| "schedule": "constant", |
| "warmup_steps": 0, |
| "weight_decay": 0.0001 |
| }, |
| "schema_version": 1, |
| "seed": 0, |
| "solver": { |
| "name": "auto", |
| "params": {} |
| }, |
| "task": { |
| "name": "regression", |
| "params": { |
| "loss": "mse", |
| "metrics": [ |
| "mse", |
| "mae" |
| ] |
| } |
| }, |
| "training": { |
| "batch_size": 256, |
| "checkpoint_interval_steps": 200, |
| "evaluation_interval_steps": 200, |
| "logging_interval_steps": 25, |
| "max_saved_checkpoints": 3, |
| "num_steps": 2000, |
| "shuffle": true, |
| "shuffle_seed": 0, |
| "steps_per_epoch": null, |
| "val_fraction": 0.1, |
| "workdir": "/Users/tr/repos/astro-emulators-toolkit/examples/development/runs/rff_mlp" |
| } |
| } |
