New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Error inpainting for CUDA #99

Open

June-phy opened this issue Mar 10, 2022 · 0 comments

June-phy commented Mar 10, 2022 •

edited

Loading

Hi,
In inpainting.ipynb the loss is NAN from the beginning.
However, if I turn off cuda:

  torch.backends.cudnn.enabled = False 
  torch.backends.cudnn.benchmark = False 
  dtype = torch.FloatTensor

All the loss is right.
I was using RTX 3070 Ti, cuda 9.0.

Why the loss turns into 'nan' when running on GPU?
Please help me with this problem.

The text was updated successfully, but these errors were encountered:

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment