epoch: 0 training_loss 0.11669733804650605 test_loss: -0.06401499509811401
