epoch: 0 training_loss nan test_loss: nan
epoch: 1 training_loss nan test_loss: nan
epoch: 2 training_loss nan test_loss: nan
