├── .gitignore ├── README.md ├── config ├── example_conf_era5.yaml └── example_conf_swe.yaml ├── data_process ├── fix_swe_data.py └── gen_SWE_from_ic_file.py ├── data_stubs ├── SWE_ICs.zip ├── era5_global_means.npy ├── era5_global_stds.npy ├── swe_global_means.npy └── swe_global_stds.npy ├── img ├── SpectralArtifacts.png ├── dfcn.gif └── fcn.gif ├── inference └── inference.py ├── networks ├── FNO.py ├── fno_utils.py ├── spectral_norm_utils.py ├── sphere_tools.py └── utilities3.py ├── ns_experiments ├── README.md ├── datasets.py ├── fno.py ├── refno.py ├── spectral_norm_utils.py └── train.py ├── train.py └── utils ├── YParams.py ├── darcy_loss.py ├── data_loader_multifiles.py ├── date_time_to_hours.py ├── img_utils.py ├── logging_utils.py ├── plots.py └── weighted_acc_rmse.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikemccabe210/stabilizing_neural_operators/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikemccabe210/stabilizing_neural_operators/HEAD/README.md -------------------------------------------------------------------------------- /config/example_conf_era5.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikemccabe210/stabilizing_neural_operators/HEAD/config/example_conf_era5.yaml -------------------------------------------------------------------------------- /config/example_conf_swe.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikemccabe210/stabilizing_neural_operators/HEAD/config/example_conf_swe.yaml -------------------------------------------------------------------------------- /data_process/fix_swe_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikemccabe210/stabilizing_neural_operators/HEAD/data_process/fix_swe_data.py -------------------------------------------------------------------------------- /data_process/gen_SWE_from_ic_file.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikemccabe210/stabilizing_neural_operators/HEAD/data_process/gen_SWE_from_ic_file.py -------------------------------------------------------------------------------- /data_stubs/SWE_ICs.zip: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikemccabe210/stabilizing_neural_operators/HEAD/data_stubs/SWE_ICs.zip -------------------------------------------------------------------------------- /data_stubs/era5_global_means.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikemccabe210/stabilizing_neural_operators/HEAD/data_stubs/era5_global_means.npy -------------------------------------------------------------------------------- /data_stubs/era5_global_stds.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikemccabe210/stabilizing_neural_operators/HEAD/data_stubs/era5_global_stds.npy -------------------------------------------------------------------------------- /data_stubs/swe_global_means.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikemccabe210/stabilizing_neural_operators/HEAD/data_stubs/swe_global_means.npy -------------------------------------------------------------------------------- /data_stubs/swe_global_stds.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikemccabe210/stabilizing_neural_operators/HEAD/data_stubs/swe_global_stds.npy -------------------------------------------------------------------------------- /img/SpectralArtifacts.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikemccabe210/stabilizing_neural_operators/HEAD/img/SpectralArtifacts.png -------------------------------------------------------------------------------- /img/dfcn.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikemccabe210/stabilizing_neural_operators/HEAD/img/dfcn.gif -------------------------------------------------------------------------------- /img/fcn.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikemccabe210/stabilizing_neural_operators/HEAD/img/fcn.gif -------------------------------------------------------------------------------- /inference/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikemccabe210/stabilizing_neural_operators/HEAD/inference/inference.py -------------------------------------------------------------------------------- /networks/FNO.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikemccabe210/stabilizing_neural_operators/HEAD/networks/FNO.py -------------------------------------------------------------------------------- /networks/fno_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikemccabe210/stabilizing_neural_operators/HEAD/networks/fno_utils.py -------------------------------------------------------------------------------- /networks/spectral_norm_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikemccabe210/stabilizing_neural_operators/HEAD/networks/spectral_norm_utils.py -------------------------------------------------------------------------------- /networks/sphere_tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikemccabe210/stabilizing_neural_operators/HEAD/networks/sphere_tools.py -------------------------------------------------------------------------------- /networks/utilities3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikemccabe210/stabilizing_neural_operators/HEAD/networks/utilities3.py -------------------------------------------------------------------------------- /ns_experiments/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikemccabe210/stabilizing_neural_operators/HEAD/ns_experiments/README.md -------------------------------------------------------------------------------- /ns_experiments/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikemccabe210/stabilizing_neural_operators/HEAD/ns_experiments/datasets.py -------------------------------------------------------------------------------- /ns_experiments/fno.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikemccabe210/stabilizing_neural_operators/HEAD/ns_experiments/fno.py -------------------------------------------------------------------------------- /ns_experiments/refno.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikemccabe210/stabilizing_neural_operators/HEAD/ns_experiments/refno.py -------------------------------------------------------------------------------- /ns_experiments/spectral_norm_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikemccabe210/stabilizing_neural_operators/HEAD/ns_experiments/spectral_norm_utils.py -------------------------------------------------------------------------------- /ns_experiments/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikemccabe210/stabilizing_neural_operators/HEAD/ns_experiments/train.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikemccabe210/stabilizing_neural_operators/HEAD/train.py -------------------------------------------------------------------------------- /utils/YParams.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikemccabe210/stabilizing_neural_operators/HEAD/utils/YParams.py -------------------------------------------------------------------------------- /utils/darcy_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikemccabe210/stabilizing_neural_operators/HEAD/utils/darcy_loss.py -------------------------------------------------------------------------------- /utils/data_loader_multifiles.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikemccabe210/stabilizing_neural_operators/HEAD/utils/data_loader_multifiles.py -------------------------------------------------------------------------------- /utils/date_time_to_hours.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikemccabe210/stabilizing_neural_operators/HEAD/utils/date_time_to_hours.py -------------------------------------------------------------------------------- /utils/img_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikemccabe210/stabilizing_neural_operators/HEAD/utils/img_utils.py -------------------------------------------------------------------------------- /utils/logging_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikemccabe210/stabilizing_neural_operators/HEAD/utils/logging_utils.py -------------------------------------------------------------------------------- /utils/plots.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikemccabe210/stabilizing_neural_operators/HEAD/utils/plots.py -------------------------------------------------------------------------------- /utils/weighted_acc_rmse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikemccabe210/stabilizing_neural_operators/HEAD/utils/weighted_acc_rmse.py --------------------------------------------------------------------------------