tatianafp
/

relevance_classifier_model

Sentence Similarity

sentence-transformers

feature-extraction

text-embeddings-inference

Model card Files Files and versions

tatianafp commited on Mar 3, 2023

Commit

d79b264

·

1 Parent(s): 195ff60

Add SetFit model

Files changed (3) hide show

README.md +3 -3
model_head.pkl +1 -1
pytorch_model.bin +1 -1

README.md CHANGED Viewed

@@ -85,7 +85,7 @@ The model was trained with the parameters:
 **DataLoader**:
-`torch.utils.data.dataloader.DataLoader` of length 34 with parameters:
 ```
 {'batch_size': 32, 'sampler': 'torch.utils.data.sampler.RandomSampler', 'batch_sampler': 'torch.utils.data.sampler.BatchSampler'}
 ```
@@ -106,8 +106,8 @@ Parameters of the fit()-Method:
         "lr": 2e-05
     },
     "scheduler": "WarmupLinear",
-    "steps_per_epoch": 34,
-    "warmup_steps": 4,
     "weight_decay": 0.01
 }
 ```

 **DataLoader**:
+`torch.utils.data.dataloader.DataLoader` of length 555 with parameters:
 ```
 {'batch_size': 32, 'sampler': 'torch.utils.data.sampler.RandomSampler', 'batch_sampler': 'torch.utils.data.sampler.BatchSampler'}
 ```
         "lr": 2e-05
     },
     "scheduler": "WarmupLinear",
+    "steps_per_epoch": 555,
+    "warmup_steps": 56,
     "weight_decay": 0.01
 }
 ```

model_head.pkl CHANGED Viewed

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:350ce8077cc72a901606395c2bf1e8b696f87f676559dd98622211be34df50c6
 size 3813

 version https://git-lfs.github.com/spec/v1
+oid sha256:9890a8e4e1a74abe1a1060aa764de5bc984e1308ba555e0bcbd2664f5964fe02
 size 3813

pytorch_model.bin CHANGED Viewed

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

 version https://git-lfs.github.com/spec/v1
+oid sha256:63c415e46fbd7bc36060589a8763c57295e19c2733c429493a8abc6a07490b68
 size 470683437