config.json

{
  "model": {
    "model_dim": 128,
    "num_attention_heads": 8,
    "num_encoder_layers": 4,
    "feedforward_dim": 256,
    "dropout_rate": 0.3,
    "attention_dropout_rate": 0.15,
    "static_embedding_dim": 64,
    "tokens_per_modality": 64,
    "max_positional_encoding": 4096,
    "num_survival_bins": 12,
    "uncertainty_samples": 5
  },
  "model_type": "cvd_ohca_predictor",
  "transformers_version": "5.14.1",
  "auto_map": {
    "AutoConfig": "configuration_ohca.OhcaConfig",
    "AutoModel": "modeling_ohca.TFCVDPredictorForOHCADetection"
  }
}