NotImplementedError #1
Description
Hello,sir! Is there any need to be modified in you demo, for which I got a error when learning about your demo. Can you help me with that?
Traceback (most recent call last):
File "full-train-cifar10.py", line 161, in
net = Net(kernel_size).cuda()
File "full-train-cifar10.py", line 119, in init
self.conv1 = SpectralParam(3, 128, kernel_size, padding=(kernel_size - 1) // 2)
File "full-train-cifar10.py", line 96, in init
weight_re, weight_im = fft.fft2(weight, torch.zeros_like(weight).cuda())
File "/home/gongyuan/anaconda3/envs/py3/lib/python3.6/site-packages/pytorch_fft/fft/fft.py", line 45, in fft2
raise NotImplementedError
NotImplementedError