code for ICML 2023 Learning Preconditioner for Conjuagate Gradient PDE Solver
Yichen Li, Peter Yichen Chen, Tao Du, Wojciech Matusik
Download sample data and place inside of the dataset folder:
https://drive.google.com/file/d/104J9-h-h0fbprC9uXiST4p814o5bSA1r/view?usp=sharing
To run a simple experiment:
bash train.sh