{
  "spec_version": 2,
  "kind": "fedformer",
  "checkpoint": "model.pypots",
  "normalization_stats": "normalization_stats.json",
  "arch": {
    "n_steps": 1440,
    "n_features": 19,
    "n_layers": 2,
    "d_model": 512,
    "n_heads": 8,
    "d_ffn": 128,
    "moving_avg_window_size": 25,
    "dropout": 0.1,
    "modes": 32,
    "mode_select": "random",
    "variant": "Fourier"
  },
  "provenance": {
    "seed": 42,
    "training": {
      "epochs": 50,
      "batch_size": 128,
      "patience": 10,
      "optimizer_lr": 0.002,
      "weight_decay": 0.0,
      "clip_grad_norm": null,
      "device": "cuda",
      "num_workers": 4
    },
    "model_config": {
      "model_name": "fedformer",
      "n_steps": 1440,
      "n_features": 19,
      "rnn_hidden_size": 128,
      "n_layers": 2,
      "d_model": 512,
      "n_heads": 8,
      "d_ffn": 128,
      "dropout": 0.1,
      "apply_nonstationary_norm": false,
      "top_k": 5,
      "n_kernels": 6,
      "moving_avg_window_size": 25,
      "individual": false,
      "ORT_weight": 1.0,
      "MIT_weight": 1.0,
      "version": "Fourier",
      "modes": 32,
      "mode_select": "random"
    },
    "h5_cache_dir": "/scratch/users/schuetzn/openmhc-imputation-train/h5/5c7dd477",
    "checkpoint_source": "/scratch/users/schuetzn/openmhc-imputation-train/runs/fedformer_20260602_031247/20260602_T031333/FEDformer.pypots"
  },
  "fourier_modes": "fourier_modes.json"
}