kowshik
/

upsc-classification-model-v1

Sentence Similarity

sentence-transformers

feature-extraction

text-embeddings-inference

Model card Files Files and versions

kowshik commited on Dec 2, 2022

Commit

db6b600

·

1 Parent(s): a4547c1

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

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

model_head.pkl CHANGED Viewed

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

 version https://git-lfs.github.com/spec/v1
+oid sha256:00aafd762c3704571a12502b08ea961b329911fc031eefec73f3837fc9193008
+size 86999

pytorch_model.bin CHANGED Viewed

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

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