In the train log, loss is always nan.
In the train log, loss is always nan.