jtwegner23
Results
2
comments of
jtwegner23
I'm having the same issue and determined the output of the second convolution of DoubleConv goes to nan during the first iteration of the down for loop. Haven't found a...
In my case it seems to be related to cuda. When I run UNET on a dataset on cpu it runs fine. When I run it on cuda it returns...