| { | |
| "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" | |
| } |