epoch: 0 training_loss 0.3329558965563774 test_loss: 0.259086537361145
