config.json

{
  "architectures": [
    "HiTrans"
  ],
  "dropout": 0.2,
  "max_doc_len": 4500,
  "max_sent_len": 256,
  "model_type": "hierarchical-transformer",
  "n_head": 8,
  "num_aux_gtsp_labels": 0,
  "num_aux_smlo_labels": 0,
  "num_aux_spp_labels": 0,
  "num_layers": 2,
  "num_main_labels": 36,
  "torch_dtype": "float32",
  "transformers_version": "4.46.0"
}