Cheng98
/

opt-125m-qnli

Text Classification

Generated from Trainer

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

Cheng98 commited on Apr 7

Commit

288d97a

•

1 Parent(s): bf32b61

Upload config

Files changed (1) hide show

config.json +6 -6

config.json CHANGED Viewed

@@ -1,5 +1,5 @@
 {
- "_name_or_path": "facebook/opt-125m",
  "_remove_final_layer_norm": false,
  "activation_dropout": 0.0,
  "activation_function": "relu",
@@ -13,16 +13,16 @@
  "enable_bias": true,
  "eos_token_id": 2,
  "ffn_dim": 3072,
- "finetuning_task": "text-classification",
  "hidden_size": 768,
  "id2label": {
- "0": "0",
- "1": "1"
  },
  "init_std": 0.02,
  "label2id": {
- "0": 0,
- "1": 1
  },
  "layer_norm_elementwise_affine": true,
  "layerdrop": 0.0,

 {
+ "_name_or_path": "./checkpoints/opt-125m-qnli",
  "_remove_final_layer_norm": false,
  "activation_dropout": 0.0,
  "activation_function": "relu",
  "enable_bias": true,
  "eos_token_id": 2,
  "ffn_dim": 3072,
+ "finetuning_task": "qnli",
  "hidden_size": 768,
  "id2label": {
+ "0": "entailment",
+ "1": "not_entailment"
  },
  "init_std": 0.02,
  "label2id": {
+ "entailment": 0,
+ "not_entailment": 1
  },
  "layer_norm_elementwise_affine": true,
  "layerdrop": 0.0,