ivanzidov
/

setfit-occupation

Sentence Similarity

sentence-transformers

feature-extraction

Inference Endpoints

Model card Files Files and versions Community

ivanzidov commited on Oct 28, 2022

Commit

588119b

·

1 Parent(s): cfe9810

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 1250 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": 1250,
-    "warmup_steps": 125,
     "weight_decay": 0.01
 }
 ```

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

model_head.pkl CHANGED Viewed

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

 version https://git-lfs.github.com/spec/v1
+oid sha256:846ef1c8c7a13cad7463074242948058e3cbcf0853b071e7141f50a1cb043357
+size 6572823

pytorch_model.bin CHANGED Viewed

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

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