search

NVlabs / stylegan2-ada-pytorch

StyleGAN2-ADA - Official PyTorch implementation
https://arxiv.org/abs/2006.06676
Other
4k stars 1.15k forks source link

Update grid_sample_gradfix.py #294

Closed afaiyaz006 closed 8 months ago

afaiyaz006 commented 8 months ago

Updated the implementation of grad_sample_gradfix to stylegan3's implementation of grad_sample_gradfix so that it works with torch version 2.1.0+cu118(The default torch version in google colab as of 10/21/2023). Please check issues: https://github.com/pytorch/pytorch/issues/34704#issuecomment-1126986323 https://github.com/NVlabs/stylegan3/issues/188.