Cuda runtime error (59)

Part 1 (2018)
26

Yep, make sure that you have a sigmoid layer at the end to get rid of negative predictions. binary_cross_entropy doesn’t like negative numbers.

2 Likes