| name: distilbert_text_classification | |
| file_name: config.yaml | |
| task: text_classification | |
| num_labels: 10 | |
| id2label: | |
| 0: LABEL_0 | |
| 1: LABEL_1 | |
| 2: LABEL_2 | |
| 3: LABEL_3 | |
| 4: LABEL_4 | |
| 5: LABEL_5 | |
| 6: LABEL_6 | |
| 7: LABEL_7 | |
| 8: LABEL_8 | |
| 9: LABEL_9 | |
| activation: gelu | |
| attention_dropout: 0.1 | |
| dim: 768 | |
| dropout: 0.1 | |
| hidden_dim: 3072 | |
| initializer_range: 0.02 | |
| max_position_embeddings: 512 | |
| model_type: distilbert | |
| n_heads: 12 | |
| n_layers: 6 | |
| output_past: true | |
| pad_token_id: 0 | |
| qa_dropout: 0.1 | |
| tie_weights_: true | |
| vocab_size: 42000 | |