Add SetFit model

README.md

@@ -85,7 +85,7 @@ The model was trained with the parameters:
 
 **DataLoader**:
 
-`torch.utils.data.dataloader.DataLoader` of length 200 with parameters:
+`torch.utils.data.dataloader.DataLoader` of length 75 with parameters:
 ```
 {'batch_size': 16, 'sampler': 'torch.utils.data.sampler.RandomSampler', 'batch_sampler': 'torch.utils.data.sampler.BatchSampler'}
 ```
@@ -97,7 +97,7 @@ The model was trained with the parameters:
 Parameters of the fit()-Method:
 ```
 {
-    "epochs": 1,
+    "epochs": 3,
     "evaluation_steps": 0,
     "evaluator": "NoneType",
     "max_grad_norm": 1,
@@ -106,8 +106,8 @@ Parameters of the fit()-Method:
         "lr": 2e-05
     },
     "scheduler": "WarmupLinear",
-    "steps_per_epoch": 200,
-    "warmup_steps": 20,
+    "steps_per_epoch": 75,
+    "warmup_steps": 8,
     "weight_decay": 0.01
 }
 ```

model_head.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d1c11d0a2f03a63f5ae276cbb9c8b3d46134c90c75bbf37dea42e6ecd435142c
+oid sha256:d0022b49ddf4d7fa4fb1a7c1ee38911d8926f078fdc43a8c449039ba2197d0a6
 size 6991

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:4bd91dc89f5fb398fbd699066c0590cb331179ecd09929858a6a92613d2f3c46
+oid sha256:59134dcce6d14050549d4becffbda54441a925fde5ed1a7b87381705fe1cba2c
 size 1112244081