Add SetFit model

README.md

@@ -46,7 +46,7 @@ The model was trained with the parameters:
 
 **DataLoader**:
 
-`torch.utils.data.dataloader.DataLoader` of length 490 with parameters:
+`torch.utils.data.dataloader.DataLoader` of length 309 with parameters:
 ```
 {'batch_size': 64, '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": 980,
-    "warmup_steps": 98,
+    "steps_per_epoch": 618,
+    "warmup_steps": 62,
     "weight_decay": 0.01
 }
 ```

model_head.pkl

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

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:9ef6b6ea2a85e0af64292373005d1831f031dd23f23d8c4d39738d9d848ce2bd
+oid sha256:c4816a7fe865b8ceb072f30c7f092c7ffc81dada5ee65c3c88672c6be1b6858a
 size 438016493