Aymen Ben Othmen commited on
Commit
69db8c0
·
1 Parent(s): 804c493

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
@@ -85,7 +85,7 @@ The model was trained with the parameters:
85
 
86
  **DataLoader**:
87
 
88
- `torch.utils.data.dataloader.DataLoader` of length 150 with parameters:
89
  ```
90
  {'batch_size': 16, 'sampler': 'torch.utils.data.sampler.RandomSampler', 'batch_sampler': 'torch.utils.data.sampler.BatchSampler'}
91
  ```
@@ -106,8 +106,8 @@ Parameters of the fit()-Method:
106
  "lr": 2e-05
107
  },
108
  "scheduler": "WarmupLinear",
109
- "steps_per_epoch": 150,
110
- "warmup_steps": 15,
111
  "weight_decay": 0.01
112
  }
113
  ```
 
85
 
86
  **DataLoader**:
87
 
88
+ `torch.utils.data.dataloader.DataLoader` of length 100 with parameters:
89
  ```
90
  {'batch_size': 16, 'sampler': 'torch.utils.data.sampler.RandomSampler', 'batch_sampler': 'torch.utils.data.sampler.BatchSampler'}
91
  ```
 
106
  "lr": 2e-05
107
  },
108
  "scheduler": "WarmupLinear",
109
+ "steps_per_epoch": 100,
110
+ "warmup_steps": 10,
111
  "weight_decay": 0.01
112
  }
113
  ```
model_head.pkl CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:00bb141ed4392b9b67a0faa2f6170797f810dee056c5bb86f6deebf58c783278
3
  size 6991
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:243599d4164e9552fb640c5286992f097c0e8b0ad993ea2e6ab52a1d25eb3211
3
  size 6991
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:2ce2553bc90a6b21fef9a73da5abaeac0726df922ae7290c3c7940e9b02e9fa2
3
  size 1112244081
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bb1e20056678baeba8ffb326364e03856703817f14ee2802daa523bcd95554ee
3
  size 1112244081