config.json

{
  "model_type": "virtual_cell_distil",
  "n_genes": 18301,
  "output_dim": 512,
  "hidden_dim": [512, 512],
  "dropout": 0.2,
  "residual": false,
  "activation": "prelu",
  "num_labels": 2,
  "classifier_dropout": 0.1,
  "auto_map": {
    "AutoConfig": "modeling_virtual_cell_distil.VirtualCellDistilConfig",
    "AutoModel": "modeling_virtual_cell_distil.VirtualCellDistilModel",
    "AutoModelForSequenceClassification": "modeling_virtual_cell_distil.VirtualCellDistilForSequenceClassification"
  }
}