This repository contains the code for DISCO.

DISCO uses a large hypernetwork (transformer) to process a short trajectory and generate the parameters of a much smaller operator network, which then predicts the next state through time integration.

With python 3.11.2.

pip install -r requirements.txt ...

python train.py --yaml-config disco_pdebench.yaml

You'll need to download PDEBench or The Well datasets and add the paths in data_utils.py.