| { | |
| "attention_heads": 64, | |
| "dropout": 0.1, | |
| "embedding_dim": 128, | |
| "encoder_layers": 15, | |
| "encoder_width": 512, | |
| "ffn_width": 2048, | |
| "gradient_checkpointing": true, | |
| "max_molecule_atoms": 256, | |
| "max_pocket_atoms": 256, | |
| "molecule_vocab_size": 31, | |
| "pocket_vocab_size": 10 | |
| } | |