seanfarrell
/

set_fit_experiment

Sentence Similarity

sentence-transformers

feature-extraction

text-embeddings-inference

Model card Files Files and versions Community

seanfarr788 commited on Nov 3, 2022

Commit

d171c1f

·

1 Parent(s): 25a7b6f

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

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

 **DataLoader**:
+`torch.utils.data.dataloader.DataLoader` of length 2035 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": 2035,
+    "warmup_steps": 204,
     "weight_decay": 0.01
 }
 ```

model_head.pkl CHANGED Viewed

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

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

pytorch_model.bin CHANGED Viewed

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

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