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