hrrsmjd/solaris_small_patch8

1{2  "model_type": "solaris-small",3  "architecture": "SolarisSmall",4  "source_repo": "hrrsmjd/solaris_small_patch8",5  "paper": "https://arxiv.org/abs/2411.16339",6  "dataset": "hrrsmjd/AIA_12hour_512x512",7  "training_steps": 7750,8  "loss": 0.007597479969263077,9  "patch_size": 8,10  "seed": 42,11  "split_scheme": "chronological_valid_samples_80_10_10_guard_24h_v1",12  "split_counts": {13    "train": 7541,14    "val": 939,15    "test": 94116  },17  "weight_decay": 0.05,18  "betas": [19    0.9,20    0.9521  ],22  "use_ema": false,23  "embed_dim": 256,24  "encoder_depths": [25    2,26    6,27    228  ],29  "decoder_depths": [30    2,31    6,32    233  ],34  "out_levels": 8,35  "wavelengths": [36    "0094",37    "0131",38    "0171",39    "0193",40    "0211",41    "0304",42    "0335",43    "1600"44  ],45  "scale_factors": [46    57.944149103084534,47    214.99738922760267,48    1590.2998402078304,49    2397.489401917806,50    1080.261734048243,51    830.778793198845,52    104.45557294825853,53    274.6568533435666454  ],55  "norm_coeff_1": 0.525581419467926,56  "norm_coeff_2": 0.4680934548377991,57  "checkpoint_file": "solaris_small_patch8_model_state_dict.pt",58  "prediction_plot": "assets/solaris_small_patch8_test0_prediction.png",59  "test_metrics_file": "eval/solaris_pretrain_p8_chronosplit_wd005_noema_test_mse_subset_0941.json",60  "test_mean_rmse_raw": 53.72720532801895,61  "test_rmse_raw": [62    2.6625704727109896,63    12.146093098231086,64    117.17874406447952,65    151.34917590125093,66    88.04643830752812,67    39.457199820072276,68    6.930603323634012,69    12.04681763624463370  ]71}72