Update README.md

README.md

@@ -41,11 +41,10 @@ Trained using NVIDIA Tesla T40 24 GB.
 
 ```
 
-  n_epochs = 3,
-  n_checkpoints = 3,
-  batch_size = 12,
-  learning_rate = 1e-5,
-
+ n_epochs = 4, # increased from 3
+ n_checkpoints = 2,
+ batch_size = 6, # decreased from 20
+ learning_rate = 1e-5,
 
 
 ```