rosinality / alias-free-gan-pytorch

Unofficial implementation of Alias-Free Generative Adversarial Networks. (https://arxiv.org/abs/2106.12423) in PyTorch
Other
507 stars 43 forks source link

Is `conv2d_gradfix` really incompatible with PyTorch > 1.8? #30

Open elimohl opened 2 years ago

elimohl commented 2 years ago

There is a check that doesn't allow to use conv2d_gradfix with PyTorch 1.9.0 (or later): https://github.com/rosinality/alias-free-gan-pytorch/blob/0e9512888128da58e2bc5945384e2212cab38ff5/stylegan2/op/conv2d_gradfix.py#L85 Is there really some specific functionality that only PyTorch 1.7 and 1.8 have, or is it the check outdated?