Add SetFit model

README.md

@@ -46,7 +46,7 @@ The model was trained with the parameters:
 
 **DataLoader**:
 
-`torch.utils.data.dataloader.DataLoader` of length 210 with parameters:
+`torch.utils.data.dataloader.DataLoader` of length 199 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": 420,
-    "warmup_steps": 42,
+    "steps_per_epoch": 398,
+    "warmup_steps": 40,
     "weight_decay": 0.01
 }
 ```

model_head.pkl

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

pytorch_model.bin

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