nickprock commited on
Commit
b26cccc
1 Parent(s): bf248c8

Upload 11 files

Browse files
Files changed (2) hide show
  1. README.md +4 -4
  2. pytorch_model.bin +1 -1
README.md CHANGED
@@ -101,16 +101,16 @@ Parameters of the fit()-Method:
101
  ```
102
  {
103
  "epochs": 10,
104
- "evaluation_steps": 0,
105
- "evaluator": "NoneType",
106
  "max_grad_norm": 1,
107
  "optimizer_class": "<class 'torch.optim.adamw.AdamW'>",
108
  "optimizer_params": {
109
  "lr": 2e-05
110
  },
111
  "scheduler": "WarmupLinear",
112
- "steps_per_epoch": null,
113
- "warmup_steps": 10000,
114
  "weight_decay": 0.01
115
  }
116
  ```
 
101
  ```
102
  {
103
  "epochs": 10,
104
+ "evaluation_steps": 500,
105
+ "evaluator": "sentence_transformers.evaluation.TripletEvaluator.TripletEvaluator",
106
  "max_grad_norm": 1,
107
  "optimizer_class": "<class 'torch.optim.adamw.AdamW'>",
108
  "optimizer_params": {
109
  "lr": 2e-05
110
  },
111
  "scheduler": "WarmupLinear",
112
+ "steps_per_epoch": 1500,
113
+ "warmup_steps": 6250,
114
  "weight_decay": 0.01
115
  }
116
  ```
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:a4e1ac687ad40d75eb4faeadfb4dad82152ba6d61af0dcf10010d7f0790cdb1a
3
  size 439781933
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2c80e62b83f1907dd603fddac86771ae7af26af8a0838007d61c0f43052d47e2
3
  size 439781933