Add SetFit model
Browse files- README.md +3 -3
- config.json +1 -1
- model_head.pkl +1 -1
- pytorch_model.bin +1 -1
- tokenizer_config.json +1 -1
README.md
CHANGED
@@ -85,7 +85,7 @@ The model was trained with the parameters:
|
|
85 |
|
86 |
**DataLoader**:
|
87 |
|
88 |
-
`torch.utils.data.dataloader.DataLoader` of length
|
89 |
```
|
90 |
{'batch_size': 64, 'sampler': 'torch.utils.data.sampler.RandomSampler', 'batch_sampler': 'torch.utils.data.sampler.BatchSampler'}
|
91 |
```
|
@@ -106,8 +106,8 @@ Parameters of the fit()-Method:
|
|
106 |
"lr": 2e-05
|
107 |
},
|
108 |
"scheduler": "WarmupLinear",
|
109 |
-
"steps_per_epoch":
|
110 |
-
"warmup_steps":
|
111 |
"weight_decay": 0.01
|
112 |
}
|
113 |
```
|
|
|
85 |
|
86 |
**DataLoader**:
|
87 |
|
88 |
+
`torch.utils.data.dataloader.DataLoader` of length 2727 with parameters:
|
89 |
```
|
90 |
{'batch_size': 64, 'sampler': 'torch.utils.data.sampler.RandomSampler', 'batch_sampler': 'torch.utils.data.sampler.BatchSampler'}
|
91 |
```
|
|
|
106 |
"lr": 2e-05
|
107 |
},
|
108 |
"scheduler": "WarmupLinear",
|
109 |
+
"steps_per_epoch": 2727,
|
110 |
+
"warmup_steps": 273,
|
111 |
"weight_decay": 0.01
|
112 |
}
|
113 |
```
|
config.json
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
{
|
2 |
-
"_name_or_path": "/home/agb/.cache/torch/sentence_transformers/
|
3 |
"architectures": [
|
4 |
"XLMRobertaModel"
|
5 |
],
|
|
|
1 |
{
|
2 |
+
"_name_or_path": "/home/agb/.cache/torch/sentence_transformers/gemasphi_real-setfit-ss-paraphrase-multilingual-mpnet-base-v2/",
|
3 |
"architectures": [
|
4 |
"XLMRobertaModel"
|
5 |
],
|
model_head.pkl
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 801640
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:eb34601b36e24928cb589c56487372a842bf0b78f6fe69d66de8b1b059fca3d3
|
3 |
size 801640
|
pytorch_model.bin
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 1112244081
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:c75a09579ca33a4684f1fee2e52810a48ef6c59a0f9fd50ebb8ef7d11d9f19af
|
3 |
size 1112244081
|
tokenizer_config.json
CHANGED
@@ -11,7 +11,7 @@
|
|
11 |
"single_word": false
|
12 |
},
|
13 |
"model_max_length": 512,
|
14 |
-
"name_or_path": "/home/agb/.cache/torch/sentence_transformers/
|
15 |
"pad_token": "<pad>",
|
16 |
"sep_token": "</s>",
|
17 |
"special_tokens_map_file": null,
|
|
|
11 |
"single_word": false
|
12 |
},
|
13 |
"model_max_length": 512,
|
14 |
+
"name_or_path": "/home/agb/.cache/torch/sentence_transformers/gemasphi_real-setfit-ss-paraphrase-multilingual-mpnet-base-v2/",
|
15 |
"pad_token": "<pad>",
|
16 |
"sep_token": "</s>",
|
17 |
"special_tokens_map_file": null,
|