Kuaaangwen commited on
Commit
a8d9205
1 Parent(s): 58d2c04

Add SetFit model

Browse files
Files changed (5) hide show
  1. README.md +3 -3
  2. config.json +1 -1
  3. model_head.pkl +1 -1
  4. pytorch_model.bin +1 -1
  5. tokenizer_config.json +1 -1
README.md CHANGED
@@ -97,7 +97,7 @@ The model was trained with the parameters:
97
  Parameters of the fit()-Method:
98
  ```
99
  {
100
- "epochs": 1,
101
  "evaluation_steps": 0,
102
  "evaluator": "NoneType",
103
  "max_grad_norm": 1,
@@ -106,8 +106,8 @@ Parameters of the fit()-Method:
106
  "lr": 2e-05
107
  },
108
  "scheduler": "WarmupLinear",
109
- "steps_per_epoch": 5470,
110
- "warmup_steps": 547,
111
  "weight_decay": 0.01
112
  }
113
  ```
 
97
  Parameters of the fit()-Method:
98
  ```
99
  {
100
+ "epochs": 2,
101
  "evaluation_steps": 0,
102
  "evaluator": "NoneType",
103
  "max_grad_norm": 1,
 
106
  "lr": 2e-05
107
  },
108
  "scheduler": "WarmupLinear",
109
+ "steps_per_epoch": 10940,
110
+ "warmup_steps": 1094,
111
  "weight_decay": 0.01
112
  }
113
  ```
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "/root/.cache/torch/sentence_transformers/sentence-transformers_paraphrase-mpnet-base-v2/",
3
  "architectures": [
4
  "MPNetModel"
5
  ],
 
1
  {
2
+ "_name_or_path": "/root/.cache/torch/sentence_transformers/Kuaaangwen_Setfit-few-shot-classifier/",
3
  "architectures": [
4
  "MPNetModel"
5
  ],
model_head.pkl CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:07cfaba4e568abb9f509316eed53627622b215fe70eb03f2f776453e865dea05
3
  size 154831
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:788ffed9b2e206e0757d8a7027564f498da3cd8ffc49b1f96c2fd0a7e2ef3218
3
  size 154831
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:2a06636857350b38c4704133a1b5d878c5bd686c0ef7d9344604823b37f5de52
3
  size 438016493
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f0316ba10cadb7ac68f486da089e8f95573e7310463ed6e81363f724f98534c5
3
  size 438016493
tokenizer_config.json CHANGED
@@ -34,7 +34,7 @@
34
  "single_word": false
35
  },
36
  "model_max_length": 512,
37
- "name_or_path": "/root/.cache/torch/sentence_transformers/sentence-transformers_paraphrase-mpnet-base-v2/",
38
  "never_split": null,
39
  "pad_token": {
40
  "__type": "AddedToken",
 
34
  "single_word": false
35
  },
36
  "model_max_length": 512,
37
+ "name_or_path": "/root/.cache/torch/sentence_transformers/Kuaaangwen_Setfit-few-shot-classifier/",
38
  "never_split": null,
39
  "pad_token": {
40
  "__type": "AddedToken",