| { | |
| "model_type": "allcaps_trihead", | |
| "architectures": [ | |
| "TransformerTriHeadLR" | |
| ], | |
| "base_model": "neuralbioinfo/prokbert-mini-long", | |
| "input_dim": 384, | |
| "output_dim": 128, | |
| "num_classes": [ | |
| 82, | |
| 53 | |
| ], | |
| "num_serotypes": 82, | |
| "num_genogroups": 53, | |
| "nhead": 4, | |
| "num_layers": 1, | |
| "max_len": 64, | |
| "chunk_size": 4000, | |
| "stride_ratio": 0.5, | |
| "max_length": 30000, | |
| "capsule_threshold": 0.5, | |
| "genogroup_head_trained": false, | |
| "torch_dtype": "float32" | |
| } |