config.json

{
    "architectures": [
        "DistilBertForSequenceClassification"
    ],
    "model_type": "distilbert",
    "vocab_size": 30522,
    "hidden_size": 768,
    "num_hidden_layers": 6,
    "num_attention_heads": 12,
    "hidden_dropout_prob": 0.1,
    "attention_probs_dropout_prob": 0.1,
    "max_position_embeddings": 512,
    "initializer_range": 0.02,
    "layer_norm_eps": 1e-12,
    "output_past": true,
    "pad_token_id": 0,
    "gradient_checkpointing": false,
    "transformers_version": "4.8.2",
    "distilbert": {
        "vocab_size": 30522,
        "dim": 768,
        "n_layers": 6,
        "n_heads": 12,
        "dropout": 0.1,
        "attention_dropout": 0.1,
        "activation": "gelu",
        "initializer_range": 0.02,
        "output_attentions": false,
        "output_hidden_states": false,
        "return_dict": true
    },
    "num_labels": 6
}