Add SetFit model

README.md

@@ -85,7 +85,7 @@ The model was trained with the parameters:
 
 **DataLoader**:
 
-`torch.utils.data.dataloader.DataLoader` of length 170 with parameters:
+`torch.utils.data.dataloader.DataLoader` of length 850 with parameters:
 ```
 {'batch_size': 8, '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": 170,
-    "warmup_steps": 17,
+    "steps_per_epoch": 850,
+    "warmup_steps": 85,
     "weight_decay": 0.01
 }
 ```

model_head.pkl

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

pytorch_model.bin

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