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