epoch: 0 training_loss 0.5557450664043426 test_loss: 0.5186703205108643
epoch: 1 training_loss 0.5198995617032051 test_loss: 0.5177731037139892
epoch: 2 training_loss 0.49166729778051377 test_loss: 0.48589267730712893
