BlueAvenir
/

sti_modern_workplace_class_model

Sentence Similarity

sentence-transformers

feature-extraction

text-embeddings-inference

Inference Endpoints

Model card Files Files and versions Community

kowshikBlue commited on May 11, 2023

Commit

cda618f

•

1 Parent(s): bb94bd5

Add 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 253 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": 253,
-    "warmup_steps": 26,
     "weight_decay": 0.01
 }
 ```

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

model_head.pkl CHANGED Viewed

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

 version https://git-lfs.github.com/spec/v1
+oid sha256:984af2fe81c39ab99a5d2039ddb4d08185735aee30080db53411b4fe954c4048
 size 6959

pytorch_model.bin CHANGED Viewed

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

 version https://git-lfs.github.com/spec/v1
+oid sha256:2d7e8ee44b8b2d7a97bd870a73d0323fda93344802e18cb8ab55fa760cfc467e
 size 1112242989