{
  "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"
  }
}