config.json

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