CUDA out of memory. #16
Description
Hello, gatheluck! Thank you for your excellent work! However, when I perform the experiments, I encounter the CUDA out of memory bug. I observe that the memory gradually increases when creating the fourier_heatmap. I conjecture that memory leakage might exist within the code, do you have any idea for solving this problem?