├── .gitignore ├── README.md ├── figs ├── flow-examples.png ├── input-output-overview.png ├── result-auto-summary.png ├── result-multistep-infer.png └── result-nonauto.png ├── generation-code ├── fluent-scheme │ ├── create_scheme.py │ ├── sample_cavity.txt │ ├── sample_karman.txt │ ├── sample_laminar.txt │ ├── sample_step.txt │ └── sample_step_geo.txt └── icem-rpl │ ├── mesh.py │ ├── project_test.rpl │ └── record_icem_rpl.md ├── requirements.txt ├── scripts ├── analysis │ ├── analysis_vae.py │ ├── analyze_cylinder_position.py │ ├── find_cylinder_centers.py │ ├── interpret_vae.py │ └── locate_cylinder.py ├── utils │ ├── get_stats.py │ └── test_vae_collapse.py └── visualization │ ├── get_result.py │ ├── plot_losses.py │ ├── plot_mask_overlay.py │ └── plot_multistep_inference.py └── src ├── .gitignore ├── args.py ├── dataset ├── __init__.py ├── base.py ├── cavity.py ├── cylinder.py ├── dam.py ├── tube.py ├── utils.py ├── vae.py └── wrapper.py ├── models ├── act_fn.py ├── auto_deeponet.py ├── auto_deeponet_cnn.py ├── auto_edeeponet.py ├── auto_ffn.py ├── base_model.py ├── cfd_vae.py ├── deeponet.py ├── ffn.py ├── fno │ ├── Adam.py │ ├── fno2d.py │ └── utilities3.py ├── gen_cast_cfd.py ├── latent_diffusion.py ├── ldm2.py ├── loss.py ├── pixel_diffusion.py ├── punetg.py ├── resnet.py └── unet.py ├── test_multistep.py ├── train.py ├── train_auto.py ├── train_diffusers.py ├── train_gencast.py ├── train_ldm.py ├── train_ldm2.py ├── train_vae.py ├── train_vae_diffsci.py ├── train_vae_kl_annealing.py └── utils ├── __init__.py ├── autoregressive.py ├── calculate_residuals_stat.py ├── common.py └── vae.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/README.md -------------------------------------------------------------------------------- /figs/flow-examples.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/figs/flow-examples.png -------------------------------------------------------------------------------- /figs/input-output-overview.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/figs/input-output-overview.png -------------------------------------------------------------------------------- /figs/result-auto-summary.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/figs/result-auto-summary.png -------------------------------------------------------------------------------- /figs/result-multistep-infer.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/figs/result-multistep-infer.png -------------------------------------------------------------------------------- /figs/result-nonauto.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/figs/result-nonauto.png -------------------------------------------------------------------------------- /generation-code/fluent-scheme/create_scheme.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/generation-code/fluent-scheme/create_scheme.py -------------------------------------------------------------------------------- /generation-code/fluent-scheme/sample_cavity.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/generation-code/fluent-scheme/sample_cavity.txt -------------------------------------------------------------------------------- /generation-code/fluent-scheme/sample_karman.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/generation-code/fluent-scheme/sample_karman.txt -------------------------------------------------------------------------------- /generation-code/fluent-scheme/sample_laminar.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/generation-code/fluent-scheme/sample_laminar.txt -------------------------------------------------------------------------------- /generation-code/fluent-scheme/sample_step.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/generation-code/fluent-scheme/sample_step.txt -------------------------------------------------------------------------------- /generation-code/fluent-scheme/sample_step_geo.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/generation-code/fluent-scheme/sample_step_geo.txt -------------------------------------------------------------------------------- /generation-code/icem-rpl/mesh.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/generation-code/icem-rpl/mesh.py -------------------------------------------------------------------------------- /generation-code/icem-rpl/project_test.rpl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/generation-code/icem-rpl/project_test.rpl -------------------------------------------------------------------------------- /generation-code/icem-rpl/record_icem_rpl.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/generation-code/icem-rpl/record_icem_rpl.md -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/analysis/analysis_vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/scripts/analysis/analysis_vae.py -------------------------------------------------------------------------------- /scripts/analysis/analyze_cylinder_position.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/scripts/analysis/analyze_cylinder_position.py -------------------------------------------------------------------------------- /scripts/analysis/find_cylinder_centers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/scripts/analysis/find_cylinder_centers.py -------------------------------------------------------------------------------- /scripts/analysis/interpret_vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/scripts/analysis/interpret_vae.py -------------------------------------------------------------------------------- /scripts/analysis/locate_cylinder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/scripts/analysis/locate_cylinder.py -------------------------------------------------------------------------------- /scripts/utils/get_stats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/scripts/utils/get_stats.py -------------------------------------------------------------------------------- /scripts/utils/test_vae_collapse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/scripts/utils/test_vae_collapse.py -------------------------------------------------------------------------------- /scripts/visualization/get_result.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/scripts/visualization/get_result.py -------------------------------------------------------------------------------- /scripts/visualization/plot_losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/scripts/visualization/plot_losses.py -------------------------------------------------------------------------------- /scripts/visualization/plot_mask_overlay.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/scripts/visualization/plot_mask_overlay.py -------------------------------------------------------------------------------- /scripts/visualization/plot_multistep_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/scripts/visualization/plot_multistep_inference.py -------------------------------------------------------------------------------- /src/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/src/.gitignore -------------------------------------------------------------------------------- /src/args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/src/args.py -------------------------------------------------------------------------------- /src/dataset/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/src/dataset/__init__.py -------------------------------------------------------------------------------- /src/dataset/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/src/dataset/base.py -------------------------------------------------------------------------------- /src/dataset/cavity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/src/dataset/cavity.py -------------------------------------------------------------------------------- /src/dataset/cylinder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/src/dataset/cylinder.py -------------------------------------------------------------------------------- /src/dataset/dam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/src/dataset/dam.py -------------------------------------------------------------------------------- /src/dataset/tube.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/src/dataset/tube.py -------------------------------------------------------------------------------- /src/dataset/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/src/dataset/utils.py -------------------------------------------------------------------------------- /src/dataset/vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/src/dataset/vae.py -------------------------------------------------------------------------------- /src/dataset/wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/src/dataset/wrapper.py -------------------------------------------------------------------------------- /src/models/act_fn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/src/models/act_fn.py -------------------------------------------------------------------------------- /src/models/auto_deeponet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/src/models/auto_deeponet.py -------------------------------------------------------------------------------- /src/models/auto_deeponet_cnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/src/models/auto_deeponet_cnn.py -------------------------------------------------------------------------------- /src/models/auto_edeeponet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/src/models/auto_edeeponet.py -------------------------------------------------------------------------------- /src/models/auto_ffn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/src/models/auto_ffn.py -------------------------------------------------------------------------------- /src/models/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/src/models/base_model.py -------------------------------------------------------------------------------- /src/models/cfd_vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/src/models/cfd_vae.py -------------------------------------------------------------------------------- /src/models/deeponet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/src/models/deeponet.py -------------------------------------------------------------------------------- /src/models/ffn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/src/models/ffn.py -------------------------------------------------------------------------------- /src/models/fno/Adam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/src/models/fno/Adam.py -------------------------------------------------------------------------------- /src/models/fno/fno2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/src/models/fno/fno2d.py -------------------------------------------------------------------------------- /src/models/fno/utilities3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/src/models/fno/utilities3.py -------------------------------------------------------------------------------- /src/models/gen_cast_cfd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/src/models/gen_cast_cfd.py -------------------------------------------------------------------------------- /src/models/latent_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/src/models/latent_diffusion.py -------------------------------------------------------------------------------- /src/models/ldm2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/src/models/ldm2.py -------------------------------------------------------------------------------- /src/models/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/src/models/loss.py -------------------------------------------------------------------------------- /src/models/pixel_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/src/models/pixel_diffusion.py -------------------------------------------------------------------------------- /src/models/punetg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/src/models/punetg.py -------------------------------------------------------------------------------- /src/models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/src/models/resnet.py -------------------------------------------------------------------------------- /src/models/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/src/models/unet.py -------------------------------------------------------------------------------- /src/test_multistep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/src/test_multistep.py -------------------------------------------------------------------------------- /src/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/src/train.py -------------------------------------------------------------------------------- /src/train_auto.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/src/train_auto.py -------------------------------------------------------------------------------- /src/train_diffusers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/src/train_diffusers.py -------------------------------------------------------------------------------- /src/train_gencast.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/src/train_gencast.py -------------------------------------------------------------------------------- /src/train_ldm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/src/train_ldm.py -------------------------------------------------------------------------------- /src/train_ldm2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/src/train_ldm2.py -------------------------------------------------------------------------------- /src/train_vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/src/train_vae.py -------------------------------------------------------------------------------- /src/train_vae_diffsci.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/src/train_vae_diffsci.py -------------------------------------------------------------------------------- /src/train_vae_kl_annealing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/src/train_vae_kl_annealing.py -------------------------------------------------------------------------------- /src/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/src/utils/__init__.py -------------------------------------------------------------------------------- /src/utils/autoregressive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/src/utils/autoregressive.py -------------------------------------------------------------------------------- /src/utils/calculate_residuals_stat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/src/utils/calculate_residuals_stat.py -------------------------------------------------------------------------------- /src/utils/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/src/utils/common.py -------------------------------------------------------------------------------- /src/utils/vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo-yining/CFDBench/HEAD/src/utils/vae.py --------------------------------------------------------------------------------