mitra-mir commited on
Commit
8cb6282
·
1 Parent(s): c7c0fa9

Add SetFit model

Browse files
Files changed (3) hide show
  1. README.md +2 -2
  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 164 with parameters:
50
  ```
51
  {'batch_size': 64, 'sampler': 'torch.utils.data.sampler.RandomSampler', 'batch_sampler': 'torch.utils.data.sampler.BatchSampler'}
52
  ```
@@ -67,7 +67,7 @@ Parameters of the fit()-Method:
67
  "lr": 2e-05
68
  },
69
  "scheduler": "WarmupLinear",
70
- "steps_per_epoch": 328,
71
  "warmup_steps": 33,
72
  "weight_decay": 0.01
73
  }
 
46
 
47
  **DataLoader**:
48
 
49
+ `torch.utils.data.dataloader.DataLoader` of length 163 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": 326,
71
  "warmup_steps": 33,
72
  "weight_decay": 0.01
73
  }
model_head.pkl CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:ea82737791af561f4fa701f7ec069afea143f29b5e34aa1ef3ff09ec084d39a9
3
  size 6991
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cb811abff6f15e31517150062e2db914ec1e1002f7c2334b3505066ae529adae
3
  size 6991
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:f69c03ea03f7da70ccfe634cf8a75e025e6947ec4dd9e102bf606d348a331937
3
  size 438016493
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a7f4775e59d1851c366eb2717b2f1496a089939c4b5b71907c01ee48ef90446f
3
  size 438016493