Add SetFit model
Browse files- README.md +3 -3
- model_head.pkl +1 -1
- pytorch_model.bin +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': 8, '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 26560 with parameters:
|
| 89 |
```
|
| 90 |
{'batch_size': 8, '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": 26560,
|
| 110 |
+
"warmup_steps": 2656,
|
| 111 |
"weight_decay": 0.01
|
| 112 |
}
|
| 113 |
```
|
model_head.pkl
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 68591
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f289f08acf9644320838fd52a5dc21cf97eff606f1a7d28a6f66a97c80f98a14
|
| 3 |
size 68591
|
pytorch_model.bin
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 438014769
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4289bc0c848b1f22d3760f6c05915bedb76f93938102b842af0bbd3116ae240b
|
| 3 |
size 438014769
|