diff options
-rw-r--r-- | lib/THNN/generic/BCECriterion.c | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/lib/THNN/generic/BCECriterion.c b/lib/THNN/generic/BCECriterion.c index 55909ba..637a406 100644 --- a/lib/THNN/generic/BCECriterion.c +++ b/lib/THNN/generic/BCECriterion.c @@ -18,12 +18,18 @@ void THNN_(BCECriterion_updateOutput)(THNNState *state, THTensor *input, real x = *input_data; real y = *target_data; real w = *weights_data; + THAssertMsg(x >= 0. && x <= 1., + "input value should be between 0~1, but got %f", + (double) x); sum -= (log(x + EPS) * y + log(1. - x + EPS) * (1. - y)) * w; ) else TH_TENSOR_APPLY2(real, input, real, target, real x = *input_data; real y = *target_data; + THAssertMsg(x >= 0. && x <= 1., + "input value should be between 0~1, but got %f", + (double) x); sum -= log(x + EPS) * y + log(1. - x + EPS) * (1. - y); ); |