epoch: 0 training_loss 6.640050139427185 test_loss: 3.5052539825439455
epoch: 1 training_loss 2.467197426557541 test_loss: 1.8064109802246093
epoch: 2 training_loss 1.117419717591256 test_loss: 0.5688773632049561
epoch: 3 training_loss 0.20538876263424755 test_loss: 0.04681603014469147
epoch: 4 training_loss -0.5317737023369409 test_loss: -0.6625177383422851
epoch: 5 training_loss -1.040359504763037 test_loss: -1.2310630798339843
epoch: 6 training_loss -1.550392518043518 test_loss: -1.9229190826416016
epoch: 7 training_loss -1.886182644367218 test_loss: -1.8908370971679687
epoch: 8 training_loss -2.255328850746155 test_loss: -2.338736724853516
epoch: 9 training_loss -2.5351985657215117 test_loss: -2.3111900329589843
epoch: 10 training_loss -2.8848093044757843 test_loss: -2.9694169998168944
epoch: 11 training_loss -2.977756600379944 test_loss: -2.934611701965332
epoch: 12 training_loss -3.1445680379867555 test_loss: -3.0658382415771483
epoch: 13 training_loss -3.3863224482536314 test_loss: -3.417360305786133
epoch: 14 training_loss -3.5524274945259093 test_loss: -3.403237533569336
epoch: 15 training_loss -3.676321356296539 test_loss: -3.571221923828125
epoch: 16 training_loss -3.8921516942977905 test_loss: -3.6383941650390623
epoch: 17 training_loss -4.039892990589141 test_loss: -3.915142059326172
epoch: 18 training_loss -4.145905923843384 test_loss: -4.0977436065673825
