Will you support pytorch>1.7.1 ? about focal-frequency-loss HOT 6 CLOSED

Thanks all for your interest and suggestions. We have supported PyTorch > 1.7.1 in this new commit. Please upgrade/reinstall the focal-frequency-loss package through pip or use the newest code in this repo.

Please close the issue if you find the problem is resolved. Thank you. :)

from focal-frequency-loss.

Is authors considering making a adaptation for high pytorch version like this :
photosynthesis-team/piq#253
Thanks,I don't familier with the fft, and want to use the great work in high pytorch version.

from focal-frequency-loss.

I was also looking into doing this transition myself, but like the other poster, am not familiar with the fft methods in Pytorch. In 1.8 the torch.fft methods were remove and torch.fft is now a full new module.

https://github.com/pytorch/pytorch/wiki/The-torch.fft-module-in-PyTorch-1.7

Most importantly: the backing code and tensors changed to be complex tensors, and so even if you replace the call to torch.rfft():

return torch.fft.fft(y, dim=2, norm='ortho')

Then later the calls to .max() fail:

from focal-frequency-loss.

@kjerk There is another unofficial implementation maybe you can refer to. https://github.com/HeonKK/Focal-Frequency-Loss/blob/main/util/ffl.py

from focal-frequency-loss.

thx

from focal-frequency-loss.

Thanks all for your interest and suggestions. We have supported PyTorch > 1.7.1 in this new commit. Please upgrade/reinstall the focal-frequency-loss package through pip or use the newest code in this repo.

Please close the issue if you find the problem is resolved. Thank you. :)

That's great news, thank you!

from focal-frequency-loss.