config.json

{
  "model_type": "solaris-small",
  "architecture": "SolarisSmall",
  "source_repo": "hrrsmjd/solaris_small_patch4",
  "paper": "https://arxiv.org/abs/2411.16339",
  "dataset": "hrrsmjd/AIA_12hour_512x512",
  "training_steps": 7750,
  "loss": 0.009133157320320606,
  "patch_size": 4,
  "seed": 42,
  "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": [
    58.224720422037755,
    216.21549287451052,
    1616.446579054541,
    2551.0149615718674,
    1190.0182024885178,
    887.1800787601859,
    112.33733897339224,
    266.61844876445224
  ],
  "norm_coeff_1": 0.5542473793029785,
  "norm_coeff_2": 0.49715369939804077,
  "checkpoint_file": "solaris_small_patch4_model_state_dict.pt",
  "prediction_plot": "assets/solaris_small_patch4_test0_prediction.png",
  "test_metrics_file": "eval/solaris_pretrain_paperloss_p4_ema_seed42_test_mse_subset_0352.json",
  "ema_available_in_training_checkpoint": true,
  "recommended_weights": "regular_final_weights"
}