Hugging Face's logo

AISE-TUDelft
/
python-parameters-classifier

Sentence Similarity
sentence-transformers
PyTorch
mpnet
feature-extraction
Model card Files
xet
Community
1
aalkaswan commited on
Commit
5b60182
·
1 Parent(s): 2ec387e

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
@@ -58,7 +58,7 @@ The model was trained with the parameters:
58
  Parameters of the fit()-Method:
59
  ```
60
  {
61
- "epochs": 6,
62
  "evaluation_steps": 0,
63
  "evaluator": "NoneType",
64
  "max_grad_norm": 1,
@@ -67,8 +67,8 @@ Parameters of the fit()-Method:
67
  "lr": 1.7094555110821448e-05
68
  },
69
  "scheduler": "WarmupLinear",
70
- "steps_per_epoch": 30558,
71
- "warmup_steps": 3056,
72
  "weight_decay": 0.01
73
  }
74
  ```
 
58
  Parameters of the fit()-Method:
59
  ```
60
  {
61
+ "epochs": 3,
62
  "evaluation_steps": 0,
63
  "evaluator": "NoneType",
64
  "max_grad_norm": 1,
 
67
  "lr": 1.7094555110821448e-05
68
  },
69
  "scheduler": "WarmupLinear",
70
+ "steps_per_epoch": 15279,
71
+ "warmup_steps": 1528,
72
  "weight_decay": 0.01
73
  }
74
  ```
model_head.pkl CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:3521bd63da5de8f3363970f085a14604722c5649fb2f0354785194109bd1903e
3
  size 6991
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b487198605d8365f779c51c386e91fc8f32e9a957f5f62cd1635459206333271
3
  size 6991
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:8439f5d147f41d4d4e65760d9d6d7275215f98b693dd374889ef069d8836e553
3
  size 438016493
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fbb46f217dd0843cdad7d37781e2d7a6ab19b9ce50d283a93f5f92eec9b38d06
3
  size 438016493