CoolFace
Modelpublic

valhalla/bart-large-sst2

sourceHugging Faceupdated 4y agoView on Hugging Face
1likes62downloads
config.json64 linesDownload Raw Back to root
1{2  "_name_or_path": "facebook-bart-large-SST-2/",3  "_num_labels": 2,4  "activation_dropout": 0.0,5  "activation_function": "gelu",6  "add_bias_logits": false,7  "add_final_layer_norm": false,8  "architectures": [9    "BartForSequenceClassification"10  ],11  "attention_dropout": 0.0,12  "bos_token_id": 0,13  "classif_dropout": 0.0,14  "classifier_dropout": 0.0,15  "d_model": 1024,16  "decoder_attention_heads": 16,17  "decoder_ffn_dim": 4096,18  "decoder_layerdrop": 0.0,19  "decoder_layers": 12,20  "decoder_start_token_id": 2,21  "dropout": 0.1,22  "encoder_attention_heads": 16,23  "encoder_ffn_dim": 4096,24  "encoder_layerdrop": 0.0,25  "encoder_layers": 12,26  "eos_token_id": 2,27  "finetuning_task": "sst-2",28  "forced_eos_token_id": 2,29  "id2label": {30    "0": "NEGATIVE",31    "1": "POSITIVE"32  },33  "init_std": 0.02,34  "is_encoder_decoder": true,35  "label2id": {36    "NEGATIVE": 0,37    "POSITIVE": 138  },39  "max_position_embeddings": 1024,40  "model_type": "bart",41  "normalize_before": false,42  "normalize_embedding": true,43  "num_hidden_layers": 12,44  "output_past": false,45  "pad_token_id": 1,46  "prefix": " ",47  "scale_embedding": false,48  "static_position_embeddings": false,49  "task_specific_params": {50    "summarization": {51      "early_stopping": true,52      "length_penalty": 2.0,53      "max_length": 142,54      "min_length": 56,55      "no_repeat_ngram_size": 3,56      "num_beams": 457    }58  },59  "torch_dtype": "float32",60  "transformers_version": "4.18.0.dev0",61  "use_cache": true,62  "vocab_size": 5026563}64