File size: 1,564 Bytes
6f07874 bd33451 6f07874 d4d3c9f 6f07874 d4d3c9f 6f07874 d4d3c9f 6f07874 d4d3c9f bd33451 d4d3c9f 6f07874 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 | {
"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
]
}
|