Callback that terminates training when a NaN loss is encountered.