huiwonLee
/

LABSE_ft_employee_sts_v1

Sentence Similarity

sentence-transformers

feature-extraction

text-embeddings-inference

Model card Files Files and versions

huiwonLee commited on May 27, 2024

Commit

c9ad4d9

·

verified ·

1 Parent(s): 6126e9c

Upload README.md with huggingface_hub

Files changed (1) hide show

README.md +3 -3

README.md CHANGED Viewed

@@ -46,7 +46,7 @@ The model was trained with the parameters:
 **DataLoader**:
-`torch.utils.data.dataloader.DataLoader` of length 23 with parameters:
 ```
 {'batch_size': 4, 'sampler': 'torch.utils.data.sampler.SequentialSampler', 'batch_sampler': 'torch.utils.data.sampler.BatchSampler'}
 ```
@@ -61,7 +61,7 @@ The model was trained with the parameters:
 Parameters of the fit()-Method:
 ```
 {
-    "epochs": 1,
     "evaluation_steps": 50,
     "evaluator": "NoneType",
     "max_grad_norm": 1,
@@ -71,7 +71,7 @@ Parameters of the fit()-Method:
     },
     "scheduler": "WarmupLinear",
     "steps_per_epoch": null,
-    "warmup_steps": 2,
     "weight_decay": 0.01
 }
 ```

 **DataLoader**:
+`torch.utils.data.dataloader.DataLoader` of length 41 with parameters:
 ```
 {'batch_size': 4, 'sampler': 'torch.utils.data.sampler.SequentialSampler', 'batch_sampler': 'torch.utils.data.sampler.BatchSampler'}
 ```
 Parameters of the fit()-Method:
 ```
 {
+    "epochs": 2,
     "evaluation_steps": 50,
     "evaluator": "NoneType",
     "max_grad_norm": 1,
     },
     "scheduler": "WarmupLinear",
     "steps_per_epoch": null,
+    "warmup_steps": 8,
     "weight_decay": 0.01
 }
 ```