GleghornLab/autoimmune_single
05
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 