This model was trained using the following environment:
Python 3.9
PyTorch 2.2.2
numpy 1.24
CUDA 12.6
comparison: Comparison of the DA-FNO with different theta and Conv-FNO and vanilla FNO models.
final_model: DA-FNO model trained on the entire training set.
dataset: The first 200 samples in the training set.
spectrum_2700: Average spectrum of the training set.