config.json

{
  "model_type": "solaris-small",
  "architecture": "SolarisSmall",
  "source_repo": "hrrsmjd/solaris_small_patch8",
  "paper": "https://arxiv.org/abs/2411.16339",
  "dataset": "hrrsmjd/AIA_12hour_512x512",
  "training_steps": 7750,
  "loss": 0.007597479969263077,
  "patch_size": 8,
  "seed": 42,
  "split_scheme": "chronological_valid_samples_80_10_10_guard_24h_v1",
  "split_counts": {
    "train": 7541,
    "val": 939,
    "test": 941
  },
  "weight_decay": 0.05,
  "betas": [
    0.9,
    0.95
  ],
  "use_ema": false,
  "embed_dim": 256,
  "encoder_depths": [
    2,
    6,
    2
  ],
  "decoder_depths": [
    2,
    6,
    2
  ],
  "out_levels": 8,
  "wavelengths": [
    "0094",
    "0131",
    "0171",
    "0193",
    "0211",
    "0304",
    "0335",
    "1600"
  ],
  "scale_factors": [
    57.944149103084534,
    214.99738922760267,
    1590.2998402078304,
    2397.489401917806,
    1080.261734048243,
    830.778793198845,
    104.45557294825853,
    274.65685334356664
  ],
  "norm_coeff_1": 0.525581419467926,
  "norm_coeff_2": 0.4680934548377991,
  "checkpoint_file": "solaris_small_patch8_model_state_dict.pt",
  "prediction_plot": "assets/solaris_small_patch8_test0_prediction.png",
  "test_metrics_file": "eval/solaris_pretrain_p8_chronosplit_wd005_noema_test_mse_subset_0941.json",
  "test_mean_rmse_raw": 53.72720532801895,
  "test_rmse_raw": [
    2.6625704727109896,
    12.146093098231086,
    117.17874406447952,
    151.34917590125093,
    88.04643830752812,
    39.457199820072276,
    6.930603323634012,
    12.046817636244633
  ]
}