epoch: 0 training_loss 0.2293594890832901 test_loss: 0.20840785503387452
