CoolFace
Modelpublic

duttaprat/HViLM-Patho

sourceHugging Faceapache-2.0updated 27d agoView on Hugging Face
0likes87downloads
config.json41 linesDownload Raw Back to root
1{2  "_name_or_path": "/data/projects/Virus/revision/hvue_v2/models/HViLM-base",3  "alibi_starting_size": 512,4  "architectures": [5    "BertForSequenceClassification"6  ],7  "attention_probs_dropout_prob": 0.0,8  "auto_map": {9    "AutoConfig": "configuration_bert.BertConfig",10    "AutoModel": "bert_layers.BertModel",11    "AutoModelForMaskedLM": "bert_layers.BertForMaskedLM",12    "AutoModelForSequenceClassification": "bert_layers.BertForSequenceClassification"13  },14  "classifier_dropout": null,15  "gradient_checkpointing": false,16  "hidden_act": "gelu",17  "hidden_dropout_prob": 0.1,18  "hidden_size": 768,19  "initializer_range": 0.02,20  "intermediate_size": 3072,21  "layer_norm_eps": 1e-12,22  "max_position_embeddings": 512,23  "num_attention_heads": 12,24  "num_hidden_layers": 12,25  "position_embedding_type": "absolute",26  "problem_type": "single_label_classification",27  "torch_dtype": "float32",28  "transformers_version": "4.28.0",29  "trust_remote_code": true,30  "type_vocab_size": 2,31  "use_cache": true,32  "vocab_size": 4096,33  "id2label": {34    "0": "NON_PATHOGENIC",35    "1": "PATHOGENIC"36  },37  "label2id": {38    "NON_PATHOGENIC": 0,39    "PATHOGENIC": 140  }41}