codegood
/

KatzBot_MiniLLM_ST

Sentence Similarity

sentence-transformers

feature-extraction

Inference Endpoints

text-embeddings-inference

Model card Files Files and versions Community

codegood commited on Feb 6

Commit

36dae3e

•

1 Parent(s): dd834ab

Add new SentenceTransformer model.

Files changed (2) hide show

README.md +3 -3
model.safetensors +1 -1

README.md CHANGED Viewed

@@ -87,7 +87,7 @@ The model was trained with the parameters:
 **DataLoader**:
-`torch.utils.data.dataloader.DataLoader` of length 211 with parameters:
 ```
 {'batch_size': 16, 'sampler': 'torch.utils.data.sampler.RandomSampler', 'batch_sampler': 'torch.utils.data.sampler.BatchSampler'}
 ```
@@ -99,7 +99,7 @@ The model was trained with the parameters:
 Parameters of the fit()-Method:
 ```
 {
-    "epochs": 60,
     "evaluation_steps": 0,
     "evaluator": "NoneType",
     "max_grad_norm": 1,
@@ -109,7 +109,7 @@ Parameters of the fit()-Method:
     },
     "scheduler": "WarmupLinear",
     "steps_per_epoch": null,
-    "warmup_steps": 1899,
     "weight_decay": 0.01
 }
 ```

 **DataLoader**:
+`torch.utils.data.dataloader.DataLoader` of length 423 with parameters:
 ```
 {'batch_size': 16, 'sampler': 'torch.utils.data.sampler.RandomSampler', 'batch_sampler': 'torch.utils.data.sampler.BatchSampler'}
 ```
 Parameters of the fit()-Method:
 ```
 {
+    "epochs": 200,
     "evaluation_steps": 0,
     "evaluator": "NoneType",
     "max_grad_norm": 1,
     },
     "scheduler": "WarmupLinear",
     "steps_per_epoch": null,
+    "warmup_steps": 12690,
     "weight_decay": 0.01
 }
 ```

model.safetensors CHANGED Viewed

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:20fa9691e2216e52e38471165cf2978f7cc189951b864ca8ef466804b292174b
 size 90864192

 version https://git-lfs.github.com/spec/v1
+oid sha256:e11df15f57619ee5c8bf01cb760b595ca3ddc5e630b3268e9765e8cf13b0580a
 size 90864192