config.json

{
  "esm_model": "facebook/esm2_t12_35M_UR50D",
  "mol_model": "seyonec/ChemBERTa-zinc-base-v1",
  "protein_dim": 480,
  "mol_dim": 768,
  "proj_dim": 256,
  "n_heads": 8,
  "dropout": 0.1,
  "freeze_encoders": true,
  "max_prot_len": 512,
  "max_mol_len": 200,
  "max_train_samples": 32,
  "max_val_samples": 16,
  "val_split": 0.05,
  "batch_size": 4,
  "lr": 0.001,
  "weight_decay": 0.01,
  "epochs": 2,
  "warmup_steps": 500,
  "grad_clip": 1.0,
  "patience": 5,
  "hub_model_id": "ethanolivertroy/MitoInteract",
  "output_dir": "/app/mitointeract_output"
}