jonkhler / s2cnn

Spherical CNNs
MIT License
939 stars 176 forks source link

Fixed breaking changes in torch.fft package in PyTorch 1.8 #51

Closed tobi-ace closed 3 years ago

tobi-ace commented 3 years ago

I fixed to functions calls to perform Fourier Transforms and Inverse Fourier transform in s2_fft and so3_fft

torch.fft and torch.ifft() functions are now called as torch.fft.fft() and torch.fft.ifft() in PyTorch 1.8

mariogeiger commented 3 years ago

Thanks, in case you only need s2 Fourier transform we have a implementation that we keep updated in e3nn