To launch the experiments, first install the requirements:
pip install -r requirements.txt
We provide wandb logs for all experiments in the paper.
Launch scripts are under scripts/EXP_NAME
. Example launch for Navier-Stokes high-viscosity
./scripts/launch/navier-stokes/high-viscosity/run_t1.sh
All datasets should be downloaded and prepared before running the scripts. The Navier-Stokes incompressible datasets from from this repository.
Instructions for ScalarFlow can be found here.
We also provide self-contained notebooks for quick experimentation on the incompressible Navier-Stokes.