File size: 1,861 Bytes
b4ffddb 3b8bc10 b4ffddb 3b8bc10 b4ffddb 3b8bc10 b4ffddb 3b8bc10 b4ffddb 3b8bc10 b4ffddb 3b8bc10 b4ffddb | 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 | {
"config": {
"train_dir": "data\\sevir_mini\\local\\train",
"val_dir": "data\\sevir_mini\\local\\val",
"output_dir": "runs\\sevir-finetuned",
"input_frames": 6,
"target_frames": 6,
"image_size": [
64,
64
],
"batch_size": 8,
"epochs": 5,
"learning_rate": 0.0001,
"base_channels": 8,
"num_workers": 0,
"seed": 42,
"device": "auto",
"amp": true,
"checkpoint_name": "rainmap-nowcasting-sevir-finetuned.safetensors",
"resume_from": "runs\\sevir-base\\rainmap-nowcasting-sevir-base.safetensors",
"dataset_name": "SEVIR VIL mini local fine-tune",
"model_warning": "Small SEVIR VIL subset fine-tuned benchmark model. Not for operational weather forecasting."
},
"device": "cpu",
"train_samples": 152,
"val_samples": 152,
"history": [
{
"epoch": 1,
"train_mse": 0.009678943915978858,
"val_mse": 0.008939616921308794,
"val_mae": 0.07144552077117719
},
{
"epoch": 2,
"train_mse": 0.008931465912610292,
"val_mse": 0.008647265770521603,
"val_mae": 0.06976475743086714
},
{
"epoch": 3,
"train_mse": 0.008475367739600571,
"val_mse": 0.008904078950811373,
"val_mae": 0.07051817798300793
},
{
"epoch": 4,
"train_mse": 0.008088804560860521,
"val_mse": 0.00873093365838653,
"val_mae": 0.06940647508752973
},
{
"epoch": 5,
"train_mse": 0.007787488449953105,
"val_mse": 0.008748720694137247,
"val_mae": 0.0691864206210563
}
],
"best_val_mse": 0.008647265770521603,
"warning": "Small SEVIR VIL subset fine-tuned benchmark model. Not for operational weather forecasting.",
"resume_from": "runs\\sevir-base\\rainmap-nowcasting-sevir-base.safetensors"
}
|