{
  "model_type": "custom_quantized",
  "architectures": ["WikiJointModel"],
  "quantization": {
    "type": "dynamic_int8",
    "enabled_layers": ["nn.Linear"],
    "backend": "fbgemm"
  },
  "config": {
    "input_dim": 2033473,
    "latent_size": 1024,
    "linker_size": 512,
    "num_categories": 11883,
    "score_threshold": 18,
    "quantized": "True",
    "quantization_type": "dynamic_int8"
  },
  "hyperparameters": {
    "learning_rate": 5e-4,
    "ae_weight": 65.0,
    "pos_weight_clamp": [1.0, 50.0],
    "dropout": 0.2,
    "cosine_spread_weight": 0.01
  },
  "training": {
    "epochs": 6,
    "batch_size": 512,
    "optimizer": "AdamW",
    "weight_decay": 1e-4
  },
  "agrs": {
    "status_dic": "status_dic",
    "pairs_list": "pairs_list",
    "ngrm_to_tri": "ngrm_to_tri",
    "id_to_text": "id_to_text",
    "text_to_id_aux": "id_to_text",
    "ngrm_id_to_text": "ngrm_id_to_text",
    "score_dic": "score_dic"
  },
  "metadata": {
    "trigram2col": "trigram2col",
    "cat2idx": "cat2idx",
    "idx2name": "idx2name",
    "id_to_text": "id_to_text",
    "text_to_id": "text_to_id"
  }
}