gaurav-mac
commited on
Commit
•
5ebb20f
1
Parent(s):
a16d736
Pushing hi sbert using sumo dataset with base l3cubepune
Browse files
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
|
89 |
```
|
90 |
{'batch_size': 16, 'sampler': 'torch.utils.data.sampler.RandomSampler', 'batch_sampler': 'torch.utils.data.sampler.BatchSampler'}
|
91 |
```
|
@@ -110,7 +110,7 @@ Parameters of the fit()-Method:
|
|
110 |
},
|
111 |
"scheduler": "WarmupLinear",
|
112 |
"steps_per_epoch": null,
|
113 |
-
"warmup_steps":
|
114 |
"weight_decay": 0.01
|
115 |
}
|
116 |
```
|
|
|
85 |
|
86 |
**DataLoader**:
|
87 |
|
88 |
+
`torch.utils.data.dataloader.DataLoader` of length 80 with parameters:
|
89 |
```
|
90 |
{'batch_size': 16, 'sampler': 'torch.utils.data.sampler.RandomSampler', 'batch_sampler': 'torch.utils.data.sampler.BatchSampler'}
|
91 |
```
|
|
|
110 |
},
|
111 |
"scheduler": "WarmupLinear",
|
112 |
"steps_per_epoch": null,
|
113 |
+
"warmup_steps": 1200,
|
114 |
"weight_decay": 0.01
|
115 |
}
|
116 |
```
|