Add SetFit model

README.md

@@ -46,7 +46,7 @@ The model was trained with the parameters:
 
 **DataLoader**:
 
-`torch.utils.data.dataloader.DataLoader` of length 78 with parameters:
+`torch.utils.data.dataloader.DataLoader` of length 59 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": 156,
-    "warmup_steps": 16,
+    "steps_per_epoch": 118,
+    "warmup_steps": 12,
     "weight_decay": 0.01
 }
 ```

model_head.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:a39628c12319cfec2ac04072610a70bf7297af5ddc7b2f0c2b877911dabd3c2d
+oid sha256:06bab0c39f708490c8f9ceac26b6acf6d4af23b01b20b6f734c30c62585e045a
 size 25463

pytorch_model.bin

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