working with complex values

[NorthernPeach]

Hello,
i'm trying to run your code but there are many errors for the complex number tensors, e.g.
RuntimeError: "batch_norm_stats_cuda" not implemented for 'ComplexFloat'

I'm curious about how you run the code. Could you share please, how shall I run it, in order to avoid these messages?

[tomer196]

Did you install the dependencies using the requirement file?
pip install -r requirement.txt

I run the code with torch==1.4. In newer versions of torch they introduce complex data type and change the FFT API. It is not complicate to adjust the code to newer torch versions if you prefer.

[NorthernPeach]

I reinstalled in again, and it works now
thank you!