| name: distilbert_text_classification | |
| config_type: model | |
| task: text_classification | |
| num_labels: 3 | |
| id2label: | |
| '0': negative | |
| '1': positive | |
| '2': neutral | |
| activation: gelu | |
| attention_dropout: 0.1 | |
| dim: 768 | |
| dropout: 0.1 | |
| hidden_dim: 3072 | |
| initializer_range: 0.02 | |
| max_position_embeddings: 512 | |
| n_heads: 12 | |
| n_layers: 6 | |
| output_past: true | |
| pad_token_id: 0 | |
| qa_dropout: 0.1 | |
| tie_weights_: true | |
| vocab_size: 42000 | |