Charul
/

my-dummy-model-5

Sentence Similarity

sentence-transformers

feature-extraction

text-embeddings-inference

Inference Endpoints

Model card Files Files and versions Community

Aymen Ben Othmen commited on Oct 7, 2022

Commit

28b13b9

·

1 Parent(s): ecff4a9

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

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

model_head.pkl CHANGED Viewed

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

 version https://git-lfs.github.com/spec/v1
+oid sha256:879b8bc82ef8f405da6ca51758ce4b7db63d4af7ad5859326b0d32ea976e6e2c
 size 6991

pytorch_model.bin CHANGED Viewed

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

 version https://git-lfs.github.com/spec/v1
+oid sha256:3a2ddb511ee1b950cd52c17d50f63984bbfeab44bc02fd4ce88bdaaa61d47062
 size 1112244081