| { | |
| "bundle": { | |
| "bundle_subdir": "bundle", | |
| "config_filename": "config.json", | |
| "metadata_filename": "metadata.json", | |
| "preprocess_filename": "preprocess.json", | |
| "weights_filename": "weights.safetensors" | |
| }, | |
| "data": { | |
| "columns": null, | |
| "dtype": "float32", | |
| "inputs": [ | |
| 0, | |
| 1, | |
| 2 | |
| ], | |
| "memmap": true, | |
| "path": "develop_data/isochrones.npy", | |
| "targets": [ | |
| 3, | |
| 4, | |
| 5 | |
| ] | |
| }, | |
| "hub": { | |
| "repo_id": null, | |
| "revision": null | |
| }, | |
| "model": { | |
| "name": "mlp", | |
| "params": { | |
| "activation": "gelu", | |
| "dtype": "float32", | |
| "hidden_sizes": [ | |
| 32, | |
| 32 | |
| ] | |
| } | |
| }, | |
| "optim": { | |
| "b1": 0.9, | |
| "b2": 0.999, | |
| "eps": 1e-08, | |
| "lr": 0.0003, | |
| "name": "adamw", | |
| "weight_decay": 0.0001 | |
| }, | |
| "preprocess": { | |
| "eps": 1e-08, | |
| "x_scaler": "standard", | |
| "y_scaler": "standard" | |
| }, | |
| "schema_version": 1, | |
| "seed": 0, | |
| "task": { | |
| "name": "regression", | |
| "params": { | |
| "loss": "mse" | |
| } | |
| }, | |
| "training": { | |
| "batch_size": 2048, | |
| "checkpoint_every_steps": 200, | |
| "eval_every_steps": 200, | |
| "log_every_steps": 25, | |
| "max_checkpoints": 3, | |
| "num_epochs": 1, | |
| "shuffle": true, | |
| "shuffle_seed": 0, | |
| "steps_per_epoch": null, | |
| "val_fraction": 0.1, | |
| "workdir": "/Users/tr/repos/astro-emulators-toolkit/examples/runs/iso_mlp" | |
| } | |
| } |