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%.
|