CoolFace
Modelpublic

GleghornLab/autoimmune_single

sourceHugging Faceupdated 2y agoView on Hugging Face
0likes5downloads
config.json59 linesDownload Raw Back to root
1{2  "ESM": false,3  "MI_loss": false,4  "MOE": false,5  "_name_or_path": "allenai/scibert_scivocab_uncased",6  "a_col": "a",7  "add_during_eval": false,8  "architectures": [9    "BertForSentenceSimilarity"10  ],11  "attention_probs_dropout_prob": 0.0,12  "b_col": "b",13  "classifier_dropout": null,14  "contact_head": false,15  "contrastive_loss": "clip",16  "data_paths": [17    "lhallee/abstract_domain_autoimmune"18  ],19  "domains": [20    "[IMM]"21  ],22  "hidden_act": "gelu",23  "hidden_dropout_prob": 0.05,24  "hidden_size": 768,25  "initializer_range": 0.02,26  "intermediate_size": 3072,27  "label_col": "label",28  "layer_norm_eps": 1e-12,29  "limits": false,30  "log_path": "./results.txt",31  "max_length": 512,32  "max_position_embeddings": 512,33  "model_path": "allenai/scibert_scivocab_uncased",34  "model_type": "bert",35  "moe_type": "topk",36  "new_special_tokens": false,37  "num_attention_heads": 12,38  "num_experts": 8,39  "num_hidden_layers": 12,40  "num_tasks": 5,41  "pad_token_id": 0,42  "patience": 3,43  "position_embedding_type": "absolute",44  "project_name": "SciMOE",45  "token_moe": false,46  "topk": 2,47  "torch_dtype": "float32",48  "transformers_version": "4.38.2",49  "type_vocab_size": 2,50  "use_cache": true,51  "vocab_size": 31090,52  "wBAL": 0.1,53  "wMI": 0.1,54  "wandb": true,55  "wandb_name": "immune_mixed_full",56  "wandb_project": "SSPR",57  "weight_path": "autoimmune_single.pt"58}59