gitextract_nppznbmv/ ├── .dockerignore ├── .gitignore ├── Dockerfile ├── LICENSE ├── README.md ├── baselines/ │ ├── __init__.py │ ├── data.py │ ├── deepxde_deeponet.py │ ├── loss.py │ ├── model.py │ ├── pinns_ns_05s.py │ ├── pinns_ns_50s.py │ ├── sapinns-50s.py │ ├── sapinns.py │ ├── test.py │ ├── tqd_sapinns.py │ ├── tqd_utils.py │ ├── train_darcy.py │ ├── train_ns.py │ ├── unet3d.py │ └── utils.py ├── cavity_flow.py ├── configs/ │ ├── baseline/ │ │ ├── NS-50s-LAAF.yaml │ │ ├── NS-50s.yaml │ │ ├── Re500-05s-deeponet.yaml │ │ ├── Re500-pinns-05s-LAAF.yaml │ │ ├── Re500-pinns-05s-SA.yaml │ │ ├── Re500-pinns-05s.yaml │ │ └── Re500-pinns.yaml │ ├── finetune/ │ │ ├── Darcy-finetune.yaml │ │ ├── Re100-finetune-1s.yaml │ │ ├── Re200-finetune-1s.yaml │ │ ├── Re250-finetune-1s.yaml │ │ ├── Re300-finetune-1s.yaml │ │ ├── Re350-finetune-1s.yaml │ │ ├── Re400-finetune-1s.yaml │ │ ├── Re500-finetune-05s-2layer.yaml │ │ ├── Re500-finetune-05s-eqn.yaml │ │ ├── Re500-finetune-05s4C0.yaml │ │ ├── Re500-finetune-05s4C1.yaml │ │ ├── Re500-finetune-05s4C4.yaml │ │ ├── Re500-finetune-05s4k-2layer.yaml │ │ ├── Re500-finetune-05s4k1k.yaml │ │ ├── Re500-finetune-05s4k4-2layer.yaml │ │ ├── Re500-finetune-05s4k4k.yaml │ │ └── Re500-finetune-1s.yaml │ ├── instance/ │ │ ├── Re500-1_8-FNO.yaml │ │ ├── Re500-1_8-PINO-s.yaml │ │ └── Re500-1_8-PINO.yaml │ ├── ngc/ │ │ ├── Re500-1_8-dat0-PINO.yaml │ │ ├── Re500-1_8-dat200-PINO.yaml │ │ ├── Re500-1_8-dat40-PINO.yaml │ │ ├── Re500-1_8-dat400-PINO.yaml │ │ ├── Re500-1_8-dat80-PINO.yaml │ │ ├── Re500-1_8-dat800-PINO.yaml │ │ ├── Re500-1_8-res16-PINO.yaml │ │ └── Re500-1_8-res32-PINO.yaml │ ├── operator/ │ │ ├── Darcy-pretrain.yaml │ │ ├── Re500-05s-1000-FNO.yaml │ │ ├── Re500-05s-1000-PINO.yaml │ │ ├── Re500-05s-3000-FNO.yaml │ │ ├── Re500-05s-600-FNO.yaml │ │ ├── Re500-05s-600-PINO-xl.yaml │ │ ├── Re500-05s-600-PINO.yaml │ │ ├── Re500-05s-FNO.yaml │ │ ├── Re500-1_16-800-FNO-s.yaml │ │ ├── Re500-1_16-800-PINO-s.yaml │ │ ├── Re500-1_4-2000-FNO.yaml │ │ ├── Re500-1_8-0-PINO-s.yaml │ │ ├── Re500-1_8-1200-FNO.yaml │ │ ├── Re500-1_8-1200-PINO.yaml │ │ ├── Re500-1_8-200-FNO-s.yaml │ │ ├── Re500-1_8-2000-FNO-s.yaml │ │ ├── Re500-1_8-2000-FNO-xl.yaml │ │ ├── Re500-1_8-2000-PINO.yaml │ │ ├── Re500-1_8-2200-FNO-s.yaml │ │ ├── Re500-1_8-2200-PINO-s.yaml │ │ ├── Re500-1_8-800-FNO-s.yaml │ │ ├── Re500-1_8-800-FNO-s32.yaml │ │ ├── Re500-1_8-800-PINO-s.yaml │ │ ├── Re500-1_8-800-PINO-s16.yaml │ │ ├── Re500-1_8-800-PINO-s32.yaml │ │ ├── Re500-1_8-800-UNet.yaml │ │ ├── Re500-1_8-dat1.6k-PINO.yaml │ │ ├── Re500-1_8-dat400-FNO.yaml │ │ ├── Re500-1s-FNO.yaml │ │ ├── Re500-3000-FNO.yaml │ │ ├── Re500-3000-PINO.yaml │ │ ├── Re500-4000-FNO.yaml │ │ ├── Re500-FNO.yaml │ │ └── Re500-PINO.yaml │ ├── pretrain/ │ │ ├── Darcy-pretrain-deeponet.yaml │ │ ├── Darcy-pretrain.yaml │ │ ├── Re100-pretrain-1s.yaml │ │ ├── Re200-pretrain-1s.yaml │ │ ├── Re250-pretrain-1s.yaml │ │ ├── Re300-pretrain-1s.yaml │ │ ├── Re350-pretrain-1s.yaml │ │ ├── Re400-pretrain-1s.yaml │ │ ├── Re500-05s-deeponet.yaml │ │ ├── Re500-FNO-1s-100.yaml │ │ ├── Re500-FNO-1s-200.yaml │ │ ├── Re500-FNO-1s-400.yaml │ │ ├── Re500-PINO-1s-100-4v4.yaml │ │ ├── Re500-PINO-1s-200-4v4.yaml │ │ ├── Re500-PINO-1s-400-1v1.yaml │ │ ├── Re500-pretrain-05s-4C1.yaml │ │ ├── Re500-pretrain-05s-4C4.yaml │ │ ├── Re500-pretrain-05s-eqn.yaml │ │ ├── Re500-pretrain-1s.yaml │ │ └── burgers-pretrain.yaml │ ├── scratch/ │ │ ├── Re100-scratch-1s.yaml │ │ ├── Re200-scratch-1s.yaml │ │ ├── Re250-scratch-1s.yaml │ │ ├── Re300-scratch-1s.yaml │ │ ├── Re350-scratch-1s.yaml │ │ ├── Re400-scratch-1s.yaml │ │ ├── Re500-scratch-05s-new.yaml │ │ ├── Re500-scratch-05s.yaml │ │ ├── Re500-scratch-1s-progressive.yaml │ │ └── Re500-scratch-1s.yaml │ ├── test/ │ │ ├── Re500-05s-deeponet.yaml │ │ ├── Re500-05s-test.yaml │ │ ├── Re500-05s.yaml │ │ ├── Re500-1s-100.yaml │ │ ├── burgers.yaml │ │ ├── darcy-deeponet.yaml │ │ └── darcy.yaml │ └── transfer/ │ ├── Re100to100-1s.yaml │ ├── Re100to200-1s.yaml │ ├── Re100to250-1s.yaml │ ├── Re100to300-1s.yaml │ ├── Re100to350-1s.yaml │ ├── Re100to400-1s.yaml │ ├── Re100to500-1s.yaml │ ├── Re200to100-1s.yaml │ ├── Re200to200-1s.yaml │ ├── Re200to250-1s.yaml │ ├── Re200to300-1s.yaml │ ├── Re200to350-1s.yaml │ ├── Re200to400-1s.yaml │ ├── Re200to500-1s.yaml │ ├── Re250to100-1s.yaml │ ├── Re250to200-1s.yaml │ ├── Re250to250-1s.yaml │ ├── Re250to300-1s.yaml │ ├── Re250to350-1s.yaml │ ├── Re250to400-1s.yaml │ ├── Re250to500-1s.yaml │ ├── Re300to100-1s.yaml │ ├── Re300to200-1s.yaml │ ├── Re300to250-1s.yaml │ ├── Re300to300-1s.yaml │ ├── Re300to350-1s.yaml │ ├── Re300to400-1s.yaml │ ├── Re300to500-1s.yaml │ ├── Re350to100-1s.yaml │ ├── Re350to200-1s.yaml │ ├── Re350to250-1s.yaml │ ├── Re350to300-1s.yaml │ ├── Re350to350-1s.yaml │ ├── Re350to400-1s.yaml │ ├── Re350to500-1s.yaml │ ├── Re400to100-1s.yaml │ ├── Re400to200-1s.yaml │ ├── Re400to250-1s.yaml │ ├── Re400to300-1s.yaml │ ├── Re400to350-1s.yaml │ ├── Re400to400-1s.yaml │ ├── Re400to500-1s.yaml │ ├── Re500to100-1s.yaml │ ├── Re500to200-1s.yaml │ ├── Re500to250-1s.yaml │ ├── Re500to300-1s.yaml │ ├── Re500to350-1s.yaml │ ├── Re500to400-1s.yaml │ ├── Re500to500-05s-new.yaml │ ├── Re500to500-05s.yaml │ └── Re500to500-1s.yaml ├── deeponet.py ├── download_data.py ├── eval_operator.py ├── generate_data.py ├── inference.py ├── instance_opt.py ├── inverse-darcy-foward.py ├── inverse-darcy.py ├── models/ │ ├── FCN.py │ ├── __init__.py │ ├── basics.py │ ├── core.py │ ├── fourier1d.py │ ├── fourier2d.py │ ├── fourier3d.py │ ├── lowrank2d.py │ ├── tfno.py │ └── utils.py ├── pinns.py ├── prepare_data.py ├── profile-solver-legacy.py ├── profiler/ │ └── calmacs.py ├── run_pino2d.py ├── run_pino3d.py ├── run_solver.py ├── scripts/ │ ├── device1-finetune.sh │ ├── device2-finetune.sh │ ├── device3.sh │ ├── finetune-4k-2layer.sh │ ├── finetune-4k0.sh │ ├── finetune-4k1-2layer.sh │ ├── finetune-4k1.sh │ ├── finetune-4k4-2layer.sh │ ├── fnoRe500.sh │ ├── ngc_submit_pino.sh │ ├── ngc_test_submit_pino.sh │ ├── pretrain.sh │ ├── scratchRe500.sh │ ├── test-opt/ │ │ └── Re500-1_8.sh │ ├── train_dat0.sh │ ├── train_dat200.sh │ ├── train_dat40.sh │ ├── train_dat400.sh │ ├── train_dat80.sh │ ├── train_dat800.sh │ ├── train_res16.sh │ └── train_res32.sh ├── solver/ │ ├── __init__.py │ ├── kolmogorov_flow.py │ ├── legacy_solver.py │ ├── periodic.py │ ├── random_fields.py │ ├── rfsampler.py │ └── spectrum.py ├── train_PINO3d.py ├── train_burgers.py ├── train_darcy.py ├── train_no.py ├── train_operator.py ├── train_pino.py ├── train_unet.py └── train_utils/ ├── __init__.py ├── adam.py ├── data_utils.py ├── datasets.py ├── distributed.py ├── eval_2d.py ├── eval_3d.py ├── losses.py ├── negadam.py ├── train_2d.py ├── train_3d.py └── utils.py