diegofernandezc
/

intropln-setfit-model

Sentence Similarity

sentence-transformers

feature-extraction

text-embeddings-inference

Model card Files Files and versions

diegofernandezc commited on Oct 8, 2022

Commit

fc12134

·

1 Parent(s): 2419302

Add SetFit model

Files changed (3) hide show

README.md +3 -3
model_head.pkl +2 -2
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 20 with parameters:
 ```
 {'batch_size': 16, '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": 20,
-    "warmup_steps": 2,
     "weight_decay": 0.01
 }
 ```

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

model_head.pkl CHANGED Viewed

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

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

pytorch_model.bin CHANGED Viewed

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

 version https://git-lfs.github.com/spec/v1
+oid sha256:924df544f563f15c73c511e6842df5943780da25e4c11bfb507539d15ab4a308
 size 438014769