| { |
| "_name_or_path": "neuralmagic/oBERT-12-upstream-pruned-unstructured-97", |
| "architectures": [ |
| "BertForSequenceClassification" |
| ], |
| "attention_probs_dropout_prob": 0.1, |
| "classifier_dropout": null, |
| "gradient_checkpointing": false, |
| "hidden_act": "gelu", |
| "hidden_dropout_prob": 0.1, |
| "hidden_size": 768, |
| "id2label": { |
| "0": "3rd party", |
| "1": "accelerator: tpu", |
| "2": "breaking change", |
| "3": "bug", |
| "4": "callback", |
| "5": "callback: model checkpoint", |
| "6": "checkpointing", |
| "7": "ci", |
| "8": "code quality", |
| "9": "data handling", |
| "10": "design", |
| "11": "discussion", |
| "12": "distributed", |
| "13": "docs", |
| "14": "duplicate", |
| "15": "environment", |
| "16": "environment: slurm", |
| "17": "example", |
| "18": "fault tolerance", |
| "19": "feature", |
| "20": "good first issue", |
| "21": "help wanted", |
| "22": "hooks", |
| "23": "let's do it!", |
| "24": "logger", |
| "25": "logger: mlflow", |
| "26": "logger: wandb", |
| "27": "logging", |
| "28": "loops", |
| "29": "lr scheduler", |
| "30": "optimizer", |
| "31": "priority: 0", |
| "32": "priority: 1", |
| "33": "priority: 2", |
| "34": "question", |
| "35": "ready", |
| "36": "refactor", |
| "37": "strategy: ddp", |
| "38": "strategy: dp", |
| "39": "trainer: test", |
| "40": "trainer: tune", |
| "41": "trainer: validate", |
| "42": "waiting on author", |
| "43": "won't fix", |
| "44": "working as intended" |
| }, |
| "initializer_range": 0.02, |
| "intermediate_size": 3072, |
| "label2id": { |
| "3rd party": 0, |
| "accelerator: tpu": 1, |
| "breaking change": 2, |
| "bug": 3, |
| "callback": 4, |
| "callback: model checkpoint": 5, |
| "checkpointing": 6, |
| "ci": 7, |
| "code quality": 8, |
| "data handling": 9, |
| "design": 10, |
| "discussion": 11, |
| "distributed": 12, |
| "docs": 13, |
| "duplicate": 14, |
| "environment": 15, |
| "environment: slurm": 16, |
| "example": 17, |
| "fault tolerance": 18, |
| "feature": 19, |
| "good first issue": 20, |
| "help wanted": 21, |
| "hooks": 22, |
| "let's do it!": 23, |
| "logger": 24, |
| "logger: mlflow": 25, |
| "logger: wandb": 26, |
| "logging": 27, |
| "loops": 28, |
| "lr scheduler": 29, |
| "optimizer": 30, |
| "priority: 0": 31, |
| "priority: 1": 32, |
| "priority: 2": 33, |
| "question": 34, |
| "ready": 35, |
| "refactor": 36, |
| "strategy: ddp": 37, |
| "strategy: dp": 38, |
| "trainer: test": 39, |
| "trainer: tune": 40, |
| "trainer: validate": 41, |
| "waiting on author": 42, |
| "won't fix": 43, |
| "working as intended": 44 |
| }, |
| "layer_norm_eps": 1e-12, |
| "max_position_embeddings": 512, |
| "model_type": "bert", |
| "num_attention_heads": 12, |
| "num_hidden_layers": 12, |
| "pad_token_id": 0, |
| "position_embedding_type": "absolute", |
| "problem_type": "multi_label_classification", |
| "torch_dtype": "float32", |
| "transformers_version": "4.25.1", |
| "type_vocab_size": 2, |
| "use_cache": true, |
| "vocab_size": 30522 |
| } |
|
|