Today we don't do any check in the inputs.
Would be handy to have some checks (NaN, Inf) before the training.