>>> Epoch 1: Perplexity: 9.383164870540728  Loss: 2.063086682881322
Loss improved inf -> 2.063086682881322
Saved training checkpoint

 >>> Epoch 2: Perplexity: 7.069309040539102  Loss: 1.884487341808205
Loss improved 2.063086682881322 -> 1.884487341808205
Saved training checkpoint

 >>> Epoch 3: Perplexity: 6.228238670911508  Loss: 1.7764355428559195
Loss improved 1.884487341808205 -> 1.7764355428559195
Saved training checkpoint

 >>> Epoch 4: Perplexity: 5.581320072936702  Loss: 1.687924663847234
Loss improved 1.7764355428559195 -> 1.687924663847234
Saved training checkpoint