Giters

andreas128 / RePaint

Official PyTorch Code and Models of "RePaint: Inpainting using Denoising Diffusion Probabilistic Models", CVPR 2022

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

NaN values for test_inet256_ex64

MaugrimEP opened this issue · comments

commented

I tried to use the test_inet256_ex64 configuration to inpaint some images that I found on the internet. But at the first time step, the values become NaN during the model forward pass. Does anyone have this issue?
gt

commented

Ok, it's an error of half-precision for conv2d. might be cuda