epoch: 0 training_loss 0.4415097925066948 test_loss: 0.3554303407669067
epoch: 1 training_loss 0.33780101299285886 test_loss: 0.2998372554779053
epoch: 2 training_loss 0.3115980465710163 test_loss: 0.2847468852996826
