Add SetFit model
Browse files- README.md +3 -3
- model_head.pkl +1 -1
- pytorch_model.bin +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":
|
| 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":
|
| 110 |
-
"warmup_steps":
|
| 111 |
"weight_decay": 0.01
|
| 112 |
}
|
| 113 |
```
|
|
|
|
| 97 |
Parameters of the fit()-Method:
|
| 98 |
```
|
| 99 |
{
|
| 100 |
+
"epochs": 10,
|
| 101 |
"evaluation_steps": 0,
|
| 102 |
"evaluator": "NoneType",
|
| 103 |
"max_grad_norm": 1,
|
|
|
|
| 106 |
"lr": 2e-05
|
| 107 |
},
|
| 108 |
"scheduler": "WarmupLinear",
|
| 109 |
+
"steps_per_epoch": 1250,
|
| 110 |
+
"warmup_steps": 125,
|
| 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 6991
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:370657d204b5b184cb00c9f2b50bac4111f9a827e3eb048d120ab85a6e5321bd
|
| 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:5ca1736a428ac56663adfe6909d822c1f2b71dcd58402421e07a361eca87802c
|
| 3 |
size 438016493
|