ModuleNotFoundError for fourier_neural_operator

[lyyc199586]

Hi!

I am trying to run train_darcy.py under ./uniform_girds:

python train_darcy.py ^
--ckpt_every 2000 ^
--iters 32000 ^
--lr 8e-4 ^
--batch_size 8 ^
--train_dataset_path ../../pde_data/Darcy_421/piececonst_r421_N1024_smooth1.mat ^
--test_dataset_path ../../pde_data/Darcy_421/piececonst_r421_N1024_smooth2.mat ^
--train_seq_num 1024 ^
--test_seq_num 100 ^
--resolution 141

But getting this module not found error:

Traceback (most recent call last):
  File "C:\workspace\no_playground\OFormer\uniform_grids\train_darcy.py", line 12, in <module>
    from nn_module.encoder_module import SpatialEncoder2D
  File "C:\workspace\no_playground\OFormer\uniform_grids\nn_module\encoder_module.py", line 10, in <module>
    from .cnn_module import PeriodicConv2d, PeriodicConv3d, UpBlock
  File "C:\workspace\no_playground\OFormer\uniform_grids\nn_module\cnn_module.py", line 8, in <module>
    from .fourier_neural_operator import SpectralConv2d_fast as fourier_conv
ModuleNotFoundError: No module named 'nn_module.fourier_neural_operator'