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
filename: /home/aryan/.hezar/models/distilbert-fa-sentiment-v1/config_1.yaml