noahjax commited on
Commit
cbcb1ee
1 Parent(s): 9a77395

Change up the categories

Browse files
Files changed (2) hide show
  1. README.md +4 -1
  2. config.json +7 -6
README.md CHANGED
@@ -1,5 +1,5 @@
1
  ---
2
- base_model: microsoft/deberta-v3-small
3
  datasets:
4
  - nyu-mll/glue
5
  - aps/super_glue
@@ -314,6 +314,9 @@ tags:
314
  - rlhf
315
  ---
316
 
 
 
 
317
  # Model Card for DeBERTa-v3-small-tasksource-nli
318
 
319
 
 
1
  ---
2
+ base_model: tasksource/deberta-small-long-nli
3
  datasets:
4
  - nyu-mll/glue
5
  - aps/super_glue
 
314
  - rlhf
315
  ---
316
 
317
+ # Details
318
+ Fork of `tasksource/deberta-small-long-nli` that changes up the default categories for easy consumption in places where configuration is otherwise difficult.
319
+
320
  # Model Card for DeBERTa-v3-small-tasksource-nli
321
 
322
 
config.json CHANGED
@@ -540,19 +540,20 @@
540
  "hidden_dropout_prob": 0.1,
541
  "hidden_size": 768,
542
  "id2label": {
543
- "0": "entailment",
544
- "1": "neutral",
545
- "2": "contradiction"
546
  },
547
  "initializer_range": 0.02,
548
  "intermediate_size": 3072,
549
  "label2id": {
550
- "contradiction": 2,
551
- "entailment": 0,
552
- "neutral": 1
553
  },
554
  "layer_norm_eps": 1e-07,
555
  "max_position_embeddings": 1680,
 
556
  "max_relative_positions": -1,
557
  "model_type": "deberta-v2",
558
  "norm_rel_ebd": "layer_norm",
 
540
  "hidden_dropout_prob": 0.1,
541
  "hidden_size": 768,
542
  "id2label": {
543
+ "0": "sports",
544
+ "1": "finance",
545
+ "2": "politics"
546
  },
547
  "initializer_range": 0.02,
548
  "intermediate_size": 3072,
549
  "label2id": {
550
+ "sports": 0,
551
+ "finance": 1,
552
+ "politics": 2
553
  },
554
  "layer_norm_eps": 1e-07,
555
  "max_position_embeddings": 1680,
556
+ "problem_type": "multi_label_classification",
557
  "max_relative_positions": -1,
558
  "model_type": "deberta-v2",
559
  "norm_rel_ebd": "layer_norm",