stylegan2-ada-pytorch icon indicating copy to clipboard operation
stylegan2-ada-pytorch copied to clipboard

Update conv2d_gradfix.py

Open angelv-salazar opened this issue 4 years ago • 0 comments

Update allow version of Pytorch to 1.10

i had this error whe run the train: stylegan conv2d_gradfix not supported on PyTorch 1.10.0+cu111

So it works without the warning when the version 1.10 is add to:

if any(torch.version.startswith(x) for x in ['1.7.', '1.8.', '1.9', '1.10.']): in line 53

angelv-salazar avatar Nov 16 '21 15:11 angelv-salazar