ModuleNotFoundError: No module named 'nn_module.fourier_neural_operator'

[RayshineRen]

It seems that the code released is not complete.
If I run 'python -m train_burgers', the error occurs:
ModuleNotFoundError: No module named 'nn_module.fourier_neural_operator'.
And there is no fourier_neural_operator.py in the folder nn_module, how can I fix this bug?

[zijieli-Jlee]

You can safely remove the import of fourier_neural_operator as it is not being used in the training of the model

[maedeh-sbri]

I have removed the 'import fourier_neural_operator', but there are still some more problems.

• In 'from dataset_new import ElectroStatData' what is dataset_new? There's no such file.

• "from utils import load_checkpoint, save_checkpoint, ensure_dir" Can you tell me the version of utils? The version that I'm using doesn't have load_checkpoint, save_checkpoint, ensure_dir.

• "from nn_module.encoder_module import IrregSpatialEncoder2D" There is no function as "IrregSpatialEncoder2D" in "encoder_module".

• "from nn_module.decoder_module import IrregSpatialDecoder2D" There is no function as "IrregSpatialDecoder2D" in "decoder_module".

How can I fix them?

[zijieli-Jlee]

Hi thanks for letting me know, dataset_new.py and utils.py are available in other directories but somehow missing in the uniform folder.

For the missing nn_module, please check https://github.com/BaratiLab/OFormer/blob/e52cd0bf269aef6757e078bb6a323cca6f221c2e/BVP/nn_module/encoder_module.py#L719C1-L720C1 and https://github.com/BaratiLab/OFormer/blob/e52cd0bf269aef6757e078bb6a323cca6f221c2e/BVP/nn_module/decoder_module.py#L1283C7-L1283C28

[maedeh-sbri]

Thanks for your answer.

But I want to run for the uniform_grids not BVP.
The links that you've provided are for the BVP. Are the decoder_module.py and ancoder_module.py the same for the uniform grids and BVP?

[zijieli-Jlee]

In uniform_grids case, irregular encoder/decoder are not necessary but if you still want to use them feel free to just copy their classes from BVP