Add SetFit model

README.md

@@ -85,9 +85,9 @@ The model was trained with the parameters:
 
 **DataLoader**:
 
-`torch.utils.data.dataloader.DataLoader` of length 8000 with parameters:
+`torch.utils.data.dataloader.DataLoader` of length 20770 with parameters:
 ```
-{'batch_size': 16, 'sampler': 'torch.utils.data.sampler.RandomSampler', 'batch_sampler': 'torch.utils.data.sampler.BatchSampler'}
+{'batch_size': 8, 'sampler': 'torch.utils.data.sampler.RandomSampler', 'batch_sampler': 'torch.utils.data.sampler.BatchSampler'}
 ```
 
 **Loss**:
@@ -106,8 +106,8 @@ Parameters of the fit()-Method:
         "lr": 2e-05
     },
     "scheduler": "WarmupLinear",
-    "steps_per_epoch": 8000,
-    "warmup_steps": 800,
+    "steps_per_epoch": 20770,
+    "warmup_steps": 2077,
     "weight_decay": 0.01
 }
 ```

model_head.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:1a8bce80975bcdd5cebd039f2268012f4aaf7238158bcd0d65aa6011c1616e46
+oid sha256:05b68228dad45f66295b34572eebb5f5c530b303ecea77771610da8b94fe5907
 size 56271

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:22b183efad6c80da795cbdb5977a04cb920206a66950a4aba0292c684f7c614c
+oid sha256:34be0da61bdaaf48bdd427ad34c8d1deb9bf99442e5aca5554308849f97c61c9
 size 438014769