mitra-mir commited on
Commit
7b23fa3
1 Parent(s): 59b68b1

Add SetFit model

Browse files
Files changed (3) hide show
  1. README.md +3 -3
  2. model_head.pkl +1 -1
  3. pytorch_model.bin +1 -1
README.md CHANGED
@@ -46,7 +46,7 @@ The model was trained with the parameters:
46
 
47
  **DataLoader**:
48
 
49
- `torch.utils.data.dataloader.DataLoader` of length 78 with parameters:
50
  ```
51
  {'batch_size': 64, 'sampler': 'torch.utils.data.sampler.RandomSampler', 'batch_sampler': 'torch.utils.data.sampler.BatchSampler'}
52
  ```
@@ -67,8 +67,8 @@ Parameters of the fit()-Method:
67
  "lr": 2e-05
68
  },
69
  "scheduler": "WarmupLinear",
70
- "steps_per_epoch": 156,
71
- "warmup_steps": 16,
72
  "weight_decay": 0.01
73
  }
74
  ```
 
46
 
47
  **DataLoader**:
48
 
49
+ `torch.utils.data.dataloader.DataLoader` of length 59 with parameters:
50
  ```
51
  {'batch_size': 64, 'sampler': 'torch.utils.data.sampler.RandomSampler', 'batch_sampler': 'torch.utils.data.sampler.BatchSampler'}
52
  ```
 
67
  "lr": 2e-05
68
  },
69
  "scheduler": "WarmupLinear",
70
+ "steps_per_epoch": 118,
71
+ "warmup_steps": 12,
72
  "weight_decay": 0.01
73
  }
74
  ```
model_head.pkl CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:a39628c12319cfec2ac04072610a70bf7297af5ddc7b2f0c2b877911dabd3c2d
3
  size 25463
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:06bab0c39f708490c8f9ceac26b6acf6d4af23b01b20b6f734c30c62585e045a
3
  size 25463
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:dc77e72b05d96b1e40607bccb515c89649e0b28aeeee541e9f96fc571bf0139f
3
  size 438016493
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c111afb65c71cff274880fd1874cf921646e1132a07e5dcdb2b1720670e53f48
3
  size 438016493