Published June 6, 2025 | Version v1
Software Open

Artifact of the paper: DiffNO: Neural Operator Learning using Physically Structured Constrained Diffusion Model

Authors/Creators

Description

The artifact includes the training and testing data and code corresponding to the Cahn-Hilliard equation in the paper titled "DiffNO: Neural Operator Learning using Physically Structured Constrained Diffusion Model." Additionally, it contains a set of pre-trained weights.

 

To train a model, first ensure that the dataset is correctly placed in the datasets folder. After setting up the environment, navigate to the folder corresponding to each experiment, modify the content in Main_Green.py, uncomment eval_stage1 and train_stage2, and run the following command to train the model:
```python
python Main_Green.py
```

To test a model, modify the content in Main_Green.py, uncomment eval_stage1 and eval_stage2, and run the following command to test the model:
```python
python Main_Green.py
```
After the program has completed all test data inference, the test results will be saved to testset_err.txt, and the average l2 error of the test set will be output, which corresponds to the results in the experiment section of the paper. The image of the predicted result versus the actual result will be saved to tesetset_pic folder.

Files

DiffNO_v1.0_CH_2d_t15_epoch400.zip

Files (250.8 MB)

Name Size Download all
md5:60e6e34d5f838f069ad4c1db2402c375
250.8 MB Preview Download