{
  "model_type": "bemae-halpha",
  "version": "1.0",
  "model_id": "anonym-submit-26/bemae-halpha-v1",
  "source_checkpoint": "runs/Halpha_all_seed42/best.pt",
  "seed": 42,
  "n_params_encoder": 794624,
  "input": {
    "n_bins": 128,
    "halpha_center_angstrom": 6562.8,
    "halpha_half_window_angstrom": 50.0,
    "wavelength_range_angstrom": [
      6512.8,
      6612.8
    ]
  },
  "output": {
    "embedding_dim": 128,
    "name": "z_halpha"
  },
  "model_config": {
    "d_model": 128,
    "n_layers": 4,
    "n_heads": 4,
    "d_ff": 512,
    "patch_size": 8,
    "patch_overlap": 4,
    "max_seq_len": 128,
    "dropout": 0.1,
    "d_decoder": 64,
    "n_decoder_layers": 2,
    "n_decoder_heads": 4
  },
  "training": {
    "val_mae_loss": 0.33542040540348916,
    "epoch_best": 45,
    "dataset_name": "anonym-submit-26/bess-bench-26",
    "subset": "Halpha_all",
    "batch_size": 256,
    "epochs": 80,
    "lr": 0.0001,
    "weight_decay": 0.05,
    "warmup_epochs": 5,
    "mask_ratio": 0.6,
    "seed": 42
  }
}