training_config.json

{
  "model_name": "SupraWeather-Nano-1.5",
  "architecture": "FT-Transformer",
  "d_token": 256,
  "n_blocks": 4,
  "n_heads": 8,
  "loss": "FocalLoss(gamma=2.0, alpha=class_balanced)",
  "optimizer": "AdamW",
  "lr": 0.001,
  "weight_decay": 0.0001,
  "scheduler": "cosine",
  "fp16": true,
  "train_samples": 240000,
  "val_samples": 30000,
  "test_samples": 30000
}