NVlabs / stylegan3

Official PyTorch implementation of StyleGAN3
Other
6.28k stars 1.1k forks source link

assertion error in _GridSample2dBackward() #623

Open Shanthika opened 8 months ago

Shanthika commented 8 months ago

Im trying to use the grid sampler implemented in stylegan3/torch_utils/ops/grid_sample_gradfix.py"

Im using the following function call: interp = grid_sample_gradfix.grid_sample(grid,coords )

However I'm getting the following error: assert_ not ctx.needs_input_grad[2] AssertionError

Can you please help me with this?