Julien21 commited on
Commit
e643cec
·
1 Parent(s): 00289a0

Add SetFit model

Browse files
Files changed (3) hide show
  1. README.md +3 -3
  2. model_head.pkl +1 -1
  3. 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": 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": 125,
110
- "warmup_steps": 13,
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:b364ba17711ffc6c8b6ea426527e35f3671685fe999f4ff423d47613a9a56a85
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:16b62f6e81055c802ffc4273aae237c3f772a79088f61223c6565ca1de2031b2
3
  size 438016493
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5ca1736a428ac56663adfe6909d822c1f2b71dcd58402421e07a361eca87802c
3
  size 438016493