labor-specter

labor-specter trained by labor_triplet

README.md

@@ -85,17 +85,14 @@ The model was trained with the parameters:
 
 **DataLoader**:
 
-`torch.utils.data.dataloader.DataLoader` of length 180000 with parameters:
+`torch.utils.data.dataloader.DataLoader` of length 5944 with parameters:
 ```
-{'batch_size': 1, 'sampler': 'torch.utils.data.sampler.RandomSampler', 'batch_sampler': 'torch.utils.data.sampler.BatchSampler'}
+{'batch_size': 4, 'sampler': 'torch.utils.data.sampler.RandomSampler', 'batch_sampler': 'torch.utils.data.sampler.BatchSampler'}
 ```
 
 **Loss**:
 
-`sentence_transformers.losses.MultipleNegativesRankingLoss.MultipleNegativesRankingLoss` with parameters:
-  ```
-  {'scale': 20.0, 'similarity_fct': 'cos_sim'}
-  ```
+`sentence_transformers.losses.CosineSimilarityLoss.CosineSimilarityLoss` 
 
 Parameters of the fit()-Method:
 ```
@@ -110,7 +107,7 @@ Parameters of the fit()-Method:
     },
     "scheduler": "WarmupLinear",
     "steps_per_epoch": null,
-    "warmup_steps": 180000,
+    "warmup_steps": 5944,
     "weight_decay": 0.01
 }
 ```