Update README.md
Browse files
README.md
CHANGED
|
@@ -144,6 +144,11 @@ recommended by the paper and 512 FFTs.
|
|
| 144 |
#### Training Hyperparameters
|
| 145 |
|
| 146 |
- **Training regime:** The model is trained in FP32
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 147 |
|
| 148 |
|
| 149 |
## Evaluation
|
|
|
|
| 144 |
#### Training Hyperparameters
|
| 145 |
|
| 146 |
- **Training regime:** The model is trained in FP32
|
| 147 |
+
- **Epochs:** 160
|
| 148 |
+
- **Batch size:** 96
|
| 149 |
+
- **Learning rate:** 0.0005
|
| 150 |
+
- **Weight decay:** 1e-6
|
| 151 |
+
- **Warmup-epochs:** 2.0
|
| 152 |
|
| 153 |
|
| 154 |
## Evaluation
|