Update README.md
Browse files
README.md
CHANGED
|
@@ -61,7 +61,7 @@ batch_size: 64
|
|
| 61 |
optimizer: adam
|
| 62 |
lr: 0.00001
|
| 63 |
weight_decay: 0
|
| 64 |
-
|
| 65 |
```
|
| 66 |
|
| 67 |
Train/validation/test splits are 80%/10%/10%.
|
|
|
|
| 61 |
optimizer: adam
|
| 62 |
lr: 0.00001
|
| 63 |
weight_decay: 0
|
| 64 |
+
epochs: 5
|
| 65 |
```
|
| 66 |
|
| 67 |
Train/validation/test splits are 80%/10%/10%.
|