Add SetFit model

README.md

@@ -85,7 +85,7 @@ The model was trained with the parameters:
 
 **DataLoader**:
 
-`torch.utils.data.dataloader.DataLoader` of length 68 with parameters:
+`torch.utils.data.dataloader.DataLoader` of length 88 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": 68,
-    "warmup_steps": 7,
+    "steps_per_epoch": 88,
+    "warmup_steps": 9,
     "weight_decay": 0.01
 }
 ```

model_head.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:b5444666ce67ab6abb8eed7a33e9be0a40b904b7ff86666cb8dcb9a886f31686
+oid sha256:07302894defe961aab768a55262e33aaa1316b0c660a1f7c0d4636022a2869ac
 size 7118

pytorch_model.bin

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