pytorch_fnet
pytorch_fnet copied to clipboard
Predict 2D images
I have noticed that there is a requirement of z-dimension >=36. I wonder where is the limitation implemented in the code? Is it possible to modify the code to make it predict 2D images?