Add SetFit model
Browse files- README.md +2 -2
- model_head.pkl +1 -1
- pytorch_model.bin +1 -1
README.md
CHANGED
|
@@ -46,7 +46,7 @@ The model was trained with the parameters:
|
|
| 46 |
|
| 47 |
**DataLoader**:
|
| 48 |
|
| 49 |
-
`torch.utils.data.dataloader.DataLoader` of length
|
| 50 |
```
|
| 51 |
{'batch_size': 64, 'sampler': 'torch.utils.data.sampler.RandomSampler', 'batch_sampler': 'torch.utils.data.sampler.BatchSampler'}
|
| 52 |
```
|
|
@@ -67,7 +67,7 @@ Parameters of the fit()-Method:
|
|
| 67 |
"lr": 2e-05
|
| 68 |
},
|
| 69 |
"scheduler": "WarmupLinear",
|
| 70 |
-
"steps_per_epoch":
|
| 71 |
"warmup_steps": 33,
|
| 72 |
"weight_decay": 0.01
|
| 73 |
}
|
|
|
|
| 46 |
|
| 47 |
**DataLoader**:
|
| 48 |
|
| 49 |
+
`torch.utils.data.dataloader.DataLoader` of length 163 with parameters:
|
| 50 |
```
|
| 51 |
{'batch_size': 64, 'sampler': 'torch.utils.data.sampler.RandomSampler', 'batch_sampler': 'torch.utils.data.sampler.BatchSampler'}
|
| 52 |
```
|
|
|
|
| 67 |
"lr": 2e-05
|
| 68 |
},
|
| 69 |
"scheduler": "WarmupLinear",
|
| 70 |
+
"steps_per_epoch": 326,
|
| 71 |
"warmup_steps": 33,
|
| 72 |
"weight_decay": 0.01
|
| 73 |
}
|
model_head.pkl
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 6991
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:cb811abff6f15e31517150062e2db914ec1e1002f7c2334b3505066ae529adae
|
| 3 |
size 6991
|
pytorch_model.bin
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 438016493
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a7f4775e59d1851c366eb2717b2f1496a089939c4b5b71907c01ee48ef90446f
|
| 3 |
size 438016493
|