training_metrics.json

{
  "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"
}