├── .gitignore ├── LICENSE ├── README.md ├── config ├── mpp_avit_L_config.yaml ├── mpp_avit_b_config.yaml ├── mpp_avit_s_config.yaml └── mpp_avit_ti_config.yaml ├── data_utils ├── datasets.py ├── hdf5_datasets.py └── mixed_dset_sampler.py ├── imgs ├── CNS_Xfer_Both.png ├── mpp_arch_v5.png └── pretraining_bars.png ├── load_env.sh ├── models ├── avit.py ├── mixed_modules.py ├── shared_modules.py ├── spatial_modules.py └── time_modules.py ├── requirements.txt ├── submit_batch.sh ├── train_basic.py └── utils ├── YParams.py └── logging_utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PolymathicAI/multiple_physics_pretraining/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PolymathicAI/multiple_physics_pretraining/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PolymathicAI/multiple_physics_pretraining/HEAD/README.md -------------------------------------------------------------------------------- /config/mpp_avit_L_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PolymathicAI/multiple_physics_pretraining/HEAD/config/mpp_avit_L_config.yaml -------------------------------------------------------------------------------- /config/mpp_avit_b_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PolymathicAI/multiple_physics_pretraining/HEAD/config/mpp_avit_b_config.yaml -------------------------------------------------------------------------------- /config/mpp_avit_s_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PolymathicAI/multiple_physics_pretraining/HEAD/config/mpp_avit_s_config.yaml -------------------------------------------------------------------------------- /config/mpp_avit_ti_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PolymathicAI/multiple_physics_pretraining/HEAD/config/mpp_avit_ti_config.yaml -------------------------------------------------------------------------------- /data_utils/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PolymathicAI/multiple_physics_pretraining/HEAD/data_utils/datasets.py -------------------------------------------------------------------------------- /data_utils/hdf5_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PolymathicAI/multiple_physics_pretraining/HEAD/data_utils/hdf5_datasets.py -------------------------------------------------------------------------------- /data_utils/mixed_dset_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PolymathicAI/multiple_physics_pretraining/HEAD/data_utils/mixed_dset_sampler.py -------------------------------------------------------------------------------- /imgs/CNS_Xfer_Both.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PolymathicAI/multiple_physics_pretraining/HEAD/imgs/CNS_Xfer_Both.png -------------------------------------------------------------------------------- /imgs/mpp_arch_v5.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PolymathicAI/multiple_physics_pretraining/HEAD/imgs/mpp_arch_v5.png -------------------------------------------------------------------------------- /imgs/pretraining_bars.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PolymathicAI/multiple_physics_pretraining/HEAD/imgs/pretraining_bars.png -------------------------------------------------------------------------------- /load_env.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PolymathicAI/multiple_physics_pretraining/HEAD/load_env.sh -------------------------------------------------------------------------------- /models/avit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PolymathicAI/multiple_physics_pretraining/HEAD/models/avit.py -------------------------------------------------------------------------------- /models/mixed_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PolymathicAI/multiple_physics_pretraining/HEAD/models/mixed_modules.py -------------------------------------------------------------------------------- /models/shared_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PolymathicAI/multiple_physics_pretraining/HEAD/models/shared_modules.py -------------------------------------------------------------------------------- /models/spatial_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PolymathicAI/multiple_physics_pretraining/HEAD/models/spatial_modules.py -------------------------------------------------------------------------------- /models/time_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PolymathicAI/multiple_physics_pretraining/HEAD/models/time_modules.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PolymathicAI/multiple_physics_pretraining/HEAD/requirements.txt -------------------------------------------------------------------------------- /submit_batch.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PolymathicAI/multiple_physics_pretraining/HEAD/submit_batch.sh -------------------------------------------------------------------------------- /train_basic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PolymathicAI/multiple_physics_pretraining/HEAD/train_basic.py -------------------------------------------------------------------------------- /utils/YParams.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PolymathicAI/multiple_physics_pretraining/HEAD/utils/YParams.py -------------------------------------------------------------------------------- /utils/logging_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PolymathicAI/multiple_physics_pretraining/HEAD/utils/logging_utils.py --------------------------------------------------------------------------------