HazyResearch / safari

Convolutions for Sequence Modeling

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

learn_ifft in long_conv.py

robflynnyh opened this issue · comments

In sequence.long_conv on line 144 there is a clause:

            if self.learn_ifft:
                y = self.block_fft_u(y_f, N=L_kernel+L,forward=False).real[..., :L]
            else:
                y = torch.fft.ifft(y_f, n=L_kernel+L, dim=-1).real[..., :L] # (B C H L)

However, learn_ifft is not defined anywhere and hence throws an error.

Great catch! A leftover option from an old version of the code :)