Help!!!
I have a problem during training. I get the error message in the title when using:
import os
os.environ['CUDA_LAUNCH_BLOCKING'] = "1"

The error is triggered by the following line going to NaN
BCE = F.binary_cross_entropy(out,real_data_I,size_average=False)
I have tried using Binary cross entropy with Logits and this gives the same error. Since the error happens at random times, I have deduced that its something to do with the input date, however i have looked at the input data its not same input data that triggers the error.
Does anyone have any ideas?
Chaslie