{
  "fold_results": [
    {
      "fold_idx": 0,
      "best_val_loss": 5.7676777839660645,
      "epochs_trained": 24,
      "final_train_loss": 1.4942118644714355
    },
    {
      "fold_idx": 1,
      "best_val_loss": 8.418675899505615,
      "epochs_trained": 20,
      "final_train_loss": 1.4902493238449097
    },
    {
      "fold_idx": 2,
      "best_val_loss": 3.5122547830854143,
      "epochs_trained": 25,
      "final_train_loss": 1.7609570423762004
    },
    {
      "fold_idx": 3,
      "best_val_loss": 3.165306806564331,
      "epochs_trained": 21,
      "final_train_loss": 2.0073827385902403
    },
    {
      "fold_idx": 4,
      "best_val_loss": 2.996154228846232,
      "epochs_trained": 18,
      "final_train_loss": 1.9732873006300493
    }
  ],
  "summary": {
    "val_loss_mean": 4.772013900393532,
    "val_loss_std": 2.0790222989111475
  },
  "config": {
    "d_model": 256,
    "num_heads": 8,
    "n_attn_layers": 4,
    "fusion_strategy": "attention",
    "head_hidden_dim": 128,
    "dropout": 0.1,
    "use_mpnn": true,
    "lr": 0.0001,
    "weight_decay": 1e-05,
    "batch_size": 32,
    "epochs": 100,
    "patience": 15,
    "init_from_pretrain": null,
    "freeze_backbone": false
  }
}