Error showing up on BCE Criterion while training

[abhinavj98]

Hey,
While training with the data you provided this error randomly shows up

/tmp/luarocks_cunn-scm-1-6757/cunn/lib/THCUNN/BCECriterion.cu:30: Acctype bce_functor<Dtype, Acctype>::operator()(Tuple) [with Tuple = thrust::detail::tuple_of_iterator_references<thrust::device_reference, thrust::device_reference, thrust::null_type, thrust::null_type, thrust::null_type, thrust::null_type, thrust::null_type, thrust::null_type, thrust::null_type, thrust::null_type>, Dtype = float, Acctype = float]: block: [0,0,0], thread: [0,0,0] Assertion input >= 0. && input <= 1. failed.
./tmp/luarocks_cunn-scm-1-6757/cunn/lib/THCUNN/BCECriterion.cu:30: Acctype bce_functor<Dtype, Acctype>::operator()(Tuple) [with Tuple = thrust::detail::tuple_of_iterator_references<thrust::device_reference, thrust::device_reference, thrust::null_type, thrust::null_type, thrust::null_type, thrust::null_type, thrust::null_type, thrust::null_type, thrust::null_type, thrust::null_type>, Dtype = float, Acctype = float]: block: [0,0,0], thread: [1,0,0] Assertion input >= 0. && input <= 1. failed.
.
.
.
/tmp/luarocks_cunn-scm-1-6757/cunn/lib/THCUNN/BCECriterion.cu:30: Acctype bce_functor<Dtype, Acctype>::operator()(Tuple) [with Tuple = thrust::detail::tuple_of_iterator_references<thrust::device_reference, thrust::device_reference, thrust::null_type, thrust::null_type, thrust::null_type, thrust::null_type, thrust::null_type, thrust::null_type, thrust::null_type, thrust::null_type>, Dtype = float, Acctype = float]: block: [0,0,0], thread: [255,0,0] Assertion input >= 0. && input <= 1. failed.

THCudaCheckWarn FAIL file=/tmp/luarocks_cutorch-scm-1-6181/cutorch/lib/THC/THCCachingHostAllocator.cpp line=211 error=59 : device-side assert triggered
THCudaCheckWarn FAIL file=/tmp/luarocks_cutorch-scm-1-6181/cutorch/lib/THC/THCCachingHostAllocator.cpp line=211 error=59 : device-side assert triggered

Is there any fix for this?

Thanks,
Abhinav

[fiftywu]

The same Question 🙂

[abhinavj98]

The same Question 🙂

Replacing BCECriterion with WeightedBCECriterion on line 339 which is defined in criterion.lua solves this issue