epoch: 0 training_loss 0.42770203709602356 test_loss: 0.36142897605895996
