| { |
| "bundle": { |
| "bundle_subdir": "bundle" |
| }, |
| "hub": { |
| "repo_id": null, |
| "revision": null |
| }, |
| "io": { |
| "x_dim": 3, |
| "x_names": [ |
| "teff", |
| "logg", |
| "feh" |
| ], |
| "y_dim": 500, |
| "y_names": [ |
| "flux" |
| ] |
| }, |
| "model": { |
| "name": "mlp", |
| "params": { |
| "activation": "gelu", |
| "dtype": "float32", |
| "hidden_sizes": [ |
| 128, |
| 128 |
| ] |
| } |
| }, |
| "optim": { |
| "b1": 0.9, |
| "b2": 0.999, |
| "decay_steps": 0, |
| "eps": 1e-08, |
| "lr": 0.001, |
| "name": "soap", |
| "precondition_1d": false, |
| "precondition_frequency": 10, |
| "schedule": "cosine", |
| "warmup_steps": 300, |
| "weight_decay": 1e-05 |
| }, |
| "schema_version": 1, |
| "seed": 0, |
| "solver": { |
| "name": "auto", |
| "params": {} |
| }, |
| "task": { |
| "name": "regression", |
| "params": { |
| "loss": "mse" |
| } |
| }, |
| "training": { |
| "batch_size": 128, |
| "checkpoint_interval_steps": 500, |
| "evaluation_interval_steps": 500, |
| "logging_interval_steps": 50, |
| "max_saved_checkpoints": 5, |
| "num_steps": 3000, |
| "shuffle": true, |
| "shuffle_seed": 0, |
| "steps_per_epoch": null, |
| "val_fraction": 0.1, |
| "workdir": "/Users/tr/repos/astro-emulators-toolkit/examples/runs/development_payne_flux_mlp_soap" |
| } |
| } |
