search

mahmoudnafifi / C5

Reference code for the paper "Cross-Camera Convolutional Color Constancy" (ICCV 2021)
Apache License 2.0
105 stars 19 forks source link

Fix rfft for PyTorch >= 1.8 #4

Closed manipopopo closed 3 years ago

manipopopo commented 3 years ago

The RFFT was changed from torch.rfft(x, 2, ...) to torch.fft.rfft2(x) instead of torch.fft.rfft in PyTorch >= 1.8.