Best validation loss: 0.7135
Step: 21
Epoch: 5