epoch: 0 training_loss 37.67618787765503 test_loss: 20.852937316894533
epoch: 1 training_loss 16.55871042251587 test_loss: 13.620852661132812
