config.json

{
 "_name_or_path": "facebook/mbart-large-50",
 "_num_labels": 3,
 "activation_dropout": 0.0,
 "activation_function": "gelu",
 "add_bias_logits": false,
 "add_final_layer_norm": true,
 "architectures": [
 "MBartForSequenceClassification"
 ],
 "attention_dropout": 0.0,
 "bos_token_id": 0,
 "classif_dropout": 0.0,
 "classifier_dropout": 0.0,
 "d_model": 1024,
 "decoder_attention_heads": 16,
 "decoder_ffn_dim": 4096,
 "decoder_layerdrop": 0.0,
 "decoder_layers": 12,
 "decoder_start_token_id": 2,
 "dropout": 0.1,
 "early_stopping": true,
 "encoder_attention_heads": 16,
 "encoder_ffn_dim": 4096,
 "encoder_layerdrop": 0.0,
 "encoder_layers": 12,
 "eos_token_id": 2,
 "forced_eos_token_id": 2,
 "gradient_checkpointing": false,
 "id2label": {
 "0": "Recreation",
 "1": "Writing",
 "2": "Entertainment",
 "3": "Private education",
 "4": "Education",
 "5": "Recreation and entertainment",
 "6": "Dance",
 "7": "Music",
 "8": "Technology",
 "9": "Science",
 "10": "Private Education",
 "11": "Debate",
 "12": "Photography",
 "13": "Coding",
 "14": "Art",
 "15": "Community",
 "16": "Sports",
 "17": "Languages",
 "18": "Culture"
 },
 "init_std": 0.02,
 "is_encoder_decoder": true,
 "label2id": {
 "Art": 14,
 "Coding": 13,
 "Community": 15,
 "Culture": 18,
 "Dance": 6,
 "Debate": 11,
 "Education": 4,
 "Entertainment": 2,
 "Languages": 17,
 "Music": 7,
 "Photography": 12,
 "Private Education": 10,
 "Private education": 3,
 "Recreation": 0,
 "Recreation and entertainment": 5,
 "Science": 9,
 "Sports": 16,
 "Technology": 8,
 "Writing": 1
 },
 "max_length": 200,
 "max_position_embeddings": 1024,
 "model_type": "mbart",
 "normalize_before": true,
 "normalize_embedding": true,
 "num_beams": 5,
 "num_hidden_layers": 12,
 "output_past": true,
 "pad_token_id": 1,
 "problem_type": "single_label_classification",
 "scale_embedding": true,
 "static_position_embeddings": false,
 "tokenizer_class": "MBart50Tokenizer",
 "torch_dtype": "float32",
 "transformers_version": "4.37.2",
 "use_cache": true,
 "vocab_size": 250054
}