epoch: 0 training_loss nan test_loss: nan
