File size: 436 Bytes
a2e25e9 a1b901f 0dd5820 2fe1f54 167b836 5114012 0dd5820 ed35673 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
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
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
file_name: config.yaml
|