Update README.md
Browse files
README.md
CHANGED
|
@@ -35,13 +35,13 @@ Midi event transformer for music generation
|
|
| 35 |
|
| 36 |
#### Training Hyperparameters
|
| 37 |
|
| 38 |
-
- lr: 2e-
|
| 39 |
- weight-decay: 0.01
|
| 40 |
-
- batch:
|
| 41 |
- fp16 mixed precision
|
| 42 |
|
| 43 |
#### Loss
|
| 44 |
-
- val loss: 0.
|
| 45 |
|
| 46 |
## Files
|
| 47 |
|
|
|
|
| 35 |
|
| 36 |
#### Training Hyperparameters
|
| 37 |
|
| 38 |
+
- lr: 2e-5
|
| 39 |
- weight-decay: 0.01
|
| 40 |
+
- batch: 4x2x4
|
| 41 |
- fp16 mixed precision
|
| 42 |
|
| 43 |
#### Loss
|
| 44 |
+
- val loss: 0.17
|
| 45 |
|
| 46 |
## Files
|
| 47 |
|