├── .gitignore ├── Airfoil-Design-AirfRANS ├── LICENSE ├── README.md ├── dataset │ ├── dataset.py │ └── dataset_stats.ipynb ├── fig │ ├── results.png │ └── task.png ├── main.py ├── main_evaluation.py ├── models │ ├── GUNet.py │ ├── GraphSAGE.py │ ├── MLP.py │ ├── NN.py │ ├── PointNet.py │ └── Transolver.py ├── params.yaml ├── requirements.txt ├── scripts │ ├── Evaluation.sh │ ├── GraphSAGE.sh │ └── Transolver.sh ├── train.py └── utils │ ├── metrics.py │ ├── metrics_NACA.py │ ├── naca_generator.py │ └── reorganize.py ├── Car-Design-ShapeNetCar ├── README.md ├── dataset │ ├── dataset.py │ └── load_dataset.py ├── fig │ ├── car_slice_surf.png │ ├── case_study.png │ ├── results.png │ └── task.png ├── main.py ├── main_evaluation.py ├── models │ └── Transolver.py ├── requirements.txt ├── scripts │ ├── Evaluation.sh │ └── Transolver.sh ├── train.py └── utils │ └── drag_coefficient.py ├── LICENSE ├── PDE-Solving-StandardBenchmark ├── README.md ├── exp_airfoil.py ├── exp_darcy.py ├── exp_elas.py ├── exp_ns.py ├── exp_pipe.py ├── exp_plas.py ├── fig │ ├── scalibility.png │ ├── showcase.png │ └── standard_benchmark.png ├── model │ ├── Embedding.py │ ├── Physics_Attention.py │ ├── Transolver_Irregular_Mesh.py │ ├── Transolver_Structured_Mesh_2D.py │ └── Transolver_Structured_Mesh_3D.py ├── model_dict.py ├── requirements.txt ├── scripts │ ├── Transolver_Airfoil.sh │ ├── Transolver_Darcy.sh │ ├── Transolver_Elas.sh │ ├── Transolver_NS.sh │ ├── Transolver_Pipe.sh │ └── Transolver_Plas.sh └── utils │ ├── normalizer.py │ └── testloss.py ├── Physics_Attention.py ├── README.md └── pic ├── Transolver.png ├── physical_states.png └── showcases.png /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/.gitignore -------------------------------------------------------------------------------- /Airfoil-Design-AirfRANS/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/Airfoil-Design-AirfRANS/LICENSE -------------------------------------------------------------------------------- /Airfoil-Design-AirfRANS/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/Airfoil-Design-AirfRANS/README.md -------------------------------------------------------------------------------- /Airfoil-Design-AirfRANS/dataset/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/Airfoil-Design-AirfRANS/dataset/dataset.py -------------------------------------------------------------------------------- /Airfoil-Design-AirfRANS/dataset/dataset_stats.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/Airfoil-Design-AirfRANS/dataset/dataset_stats.ipynb -------------------------------------------------------------------------------- /Airfoil-Design-AirfRANS/fig/results.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/Airfoil-Design-AirfRANS/fig/results.png -------------------------------------------------------------------------------- /Airfoil-Design-AirfRANS/fig/task.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/Airfoil-Design-AirfRANS/fig/task.png -------------------------------------------------------------------------------- /Airfoil-Design-AirfRANS/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/Airfoil-Design-AirfRANS/main.py -------------------------------------------------------------------------------- /Airfoil-Design-AirfRANS/main_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/Airfoil-Design-AirfRANS/main_evaluation.py -------------------------------------------------------------------------------- /Airfoil-Design-AirfRANS/models/GUNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/Airfoil-Design-AirfRANS/models/GUNet.py -------------------------------------------------------------------------------- /Airfoil-Design-AirfRANS/models/GraphSAGE.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/Airfoil-Design-AirfRANS/models/GraphSAGE.py -------------------------------------------------------------------------------- /Airfoil-Design-AirfRANS/models/MLP.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/Airfoil-Design-AirfRANS/models/MLP.py -------------------------------------------------------------------------------- /Airfoil-Design-AirfRANS/models/NN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/Airfoil-Design-AirfRANS/models/NN.py -------------------------------------------------------------------------------- /Airfoil-Design-AirfRANS/models/PointNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/Airfoil-Design-AirfRANS/models/PointNet.py -------------------------------------------------------------------------------- /Airfoil-Design-AirfRANS/models/Transolver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/Airfoil-Design-AirfRANS/models/Transolver.py -------------------------------------------------------------------------------- /Airfoil-Design-AirfRANS/params.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/Airfoil-Design-AirfRANS/params.yaml -------------------------------------------------------------------------------- /Airfoil-Design-AirfRANS/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/Airfoil-Design-AirfRANS/requirements.txt -------------------------------------------------------------------------------- /Airfoil-Design-AirfRANS/scripts/Evaluation.sh: -------------------------------------------------------------------------------- 1 | export CUDA_VISIBLE_DEVICES=4 2 | 3 | python main_evaluation.py --my_path /data/naca/ -------------------------------------------------------------------------------- /Airfoil-Design-AirfRANS/scripts/GraphSAGE.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/Airfoil-Design-AirfRANS/scripts/GraphSAGE.sh -------------------------------------------------------------------------------- /Airfoil-Design-AirfRANS/scripts/Transolver.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/Airfoil-Design-AirfRANS/scripts/Transolver.sh -------------------------------------------------------------------------------- /Airfoil-Design-AirfRANS/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/Airfoil-Design-AirfRANS/train.py -------------------------------------------------------------------------------- /Airfoil-Design-AirfRANS/utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/Airfoil-Design-AirfRANS/utils/metrics.py -------------------------------------------------------------------------------- /Airfoil-Design-AirfRANS/utils/metrics_NACA.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/Airfoil-Design-AirfRANS/utils/metrics_NACA.py -------------------------------------------------------------------------------- /Airfoil-Design-AirfRANS/utils/naca_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/Airfoil-Design-AirfRANS/utils/naca_generator.py -------------------------------------------------------------------------------- /Airfoil-Design-AirfRANS/utils/reorganize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/Airfoil-Design-AirfRANS/utils/reorganize.py -------------------------------------------------------------------------------- /Car-Design-ShapeNetCar/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/Car-Design-ShapeNetCar/README.md -------------------------------------------------------------------------------- /Car-Design-ShapeNetCar/dataset/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/Car-Design-ShapeNetCar/dataset/dataset.py -------------------------------------------------------------------------------- /Car-Design-ShapeNetCar/dataset/load_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/Car-Design-ShapeNetCar/dataset/load_dataset.py -------------------------------------------------------------------------------- /Car-Design-ShapeNetCar/fig/car_slice_surf.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/Car-Design-ShapeNetCar/fig/car_slice_surf.png -------------------------------------------------------------------------------- /Car-Design-ShapeNetCar/fig/case_study.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/Car-Design-ShapeNetCar/fig/case_study.png -------------------------------------------------------------------------------- /Car-Design-ShapeNetCar/fig/results.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/Car-Design-ShapeNetCar/fig/results.png -------------------------------------------------------------------------------- /Car-Design-ShapeNetCar/fig/task.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/Car-Design-ShapeNetCar/fig/task.png -------------------------------------------------------------------------------- /Car-Design-ShapeNetCar/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/Car-Design-ShapeNetCar/main.py -------------------------------------------------------------------------------- /Car-Design-ShapeNetCar/main_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/Car-Design-ShapeNetCar/main_evaluation.py -------------------------------------------------------------------------------- /Car-Design-ShapeNetCar/models/Transolver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/Car-Design-ShapeNetCar/models/Transolver.py -------------------------------------------------------------------------------- /Car-Design-ShapeNetCar/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/Car-Design-ShapeNetCar/requirements.txt -------------------------------------------------------------------------------- /Car-Design-ShapeNetCar/scripts/Evaluation.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/Car-Design-ShapeNetCar/scripts/Evaluation.sh -------------------------------------------------------------------------------- /Car-Design-ShapeNetCar/scripts/Transolver.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/Car-Design-ShapeNetCar/scripts/Transolver.sh -------------------------------------------------------------------------------- /Car-Design-ShapeNetCar/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/Car-Design-ShapeNetCar/train.py -------------------------------------------------------------------------------- /Car-Design-ShapeNetCar/utils/drag_coefficient.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/Car-Design-ShapeNetCar/utils/drag_coefficient.py -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/LICENSE -------------------------------------------------------------------------------- /PDE-Solving-StandardBenchmark/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/PDE-Solving-StandardBenchmark/README.md -------------------------------------------------------------------------------- /PDE-Solving-StandardBenchmark/exp_airfoil.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/PDE-Solving-StandardBenchmark/exp_airfoil.py -------------------------------------------------------------------------------- /PDE-Solving-StandardBenchmark/exp_darcy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/PDE-Solving-StandardBenchmark/exp_darcy.py -------------------------------------------------------------------------------- /PDE-Solving-StandardBenchmark/exp_elas.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/PDE-Solving-StandardBenchmark/exp_elas.py -------------------------------------------------------------------------------- /PDE-Solving-StandardBenchmark/exp_ns.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/PDE-Solving-StandardBenchmark/exp_ns.py -------------------------------------------------------------------------------- /PDE-Solving-StandardBenchmark/exp_pipe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/PDE-Solving-StandardBenchmark/exp_pipe.py -------------------------------------------------------------------------------- /PDE-Solving-StandardBenchmark/exp_plas.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/PDE-Solving-StandardBenchmark/exp_plas.py -------------------------------------------------------------------------------- /PDE-Solving-StandardBenchmark/fig/scalibility.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/PDE-Solving-StandardBenchmark/fig/scalibility.png -------------------------------------------------------------------------------- /PDE-Solving-StandardBenchmark/fig/showcase.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/PDE-Solving-StandardBenchmark/fig/showcase.png -------------------------------------------------------------------------------- /PDE-Solving-StandardBenchmark/fig/standard_benchmark.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/PDE-Solving-StandardBenchmark/fig/standard_benchmark.png -------------------------------------------------------------------------------- /PDE-Solving-StandardBenchmark/model/Embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/PDE-Solving-StandardBenchmark/model/Embedding.py -------------------------------------------------------------------------------- /PDE-Solving-StandardBenchmark/model/Physics_Attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/PDE-Solving-StandardBenchmark/model/Physics_Attention.py -------------------------------------------------------------------------------- /PDE-Solving-StandardBenchmark/model/Transolver_Irregular_Mesh.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/PDE-Solving-StandardBenchmark/model/Transolver_Irregular_Mesh.py -------------------------------------------------------------------------------- /PDE-Solving-StandardBenchmark/model/Transolver_Structured_Mesh_2D.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/PDE-Solving-StandardBenchmark/model/Transolver_Structured_Mesh_2D.py -------------------------------------------------------------------------------- /PDE-Solving-StandardBenchmark/model/Transolver_Structured_Mesh_3D.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/PDE-Solving-StandardBenchmark/model/Transolver_Structured_Mesh_3D.py -------------------------------------------------------------------------------- /PDE-Solving-StandardBenchmark/model_dict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/PDE-Solving-StandardBenchmark/model_dict.py -------------------------------------------------------------------------------- /PDE-Solving-StandardBenchmark/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/PDE-Solving-StandardBenchmark/requirements.txt -------------------------------------------------------------------------------- /PDE-Solving-StandardBenchmark/scripts/Transolver_Airfoil.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/PDE-Solving-StandardBenchmark/scripts/Transolver_Airfoil.sh -------------------------------------------------------------------------------- /PDE-Solving-StandardBenchmark/scripts/Transolver_Darcy.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/PDE-Solving-StandardBenchmark/scripts/Transolver_Darcy.sh -------------------------------------------------------------------------------- /PDE-Solving-StandardBenchmark/scripts/Transolver_Elas.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/PDE-Solving-StandardBenchmark/scripts/Transolver_Elas.sh -------------------------------------------------------------------------------- /PDE-Solving-StandardBenchmark/scripts/Transolver_NS.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/PDE-Solving-StandardBenchmark/scripts/Transolver_NS.sh -------------------------------------------------------------------------------- /PDE-Solving-StandardBenchmark/scripts/Transolver_Pipe.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/PDE-Solving-StandardBenchmark/scripts/Transolver_Pipe.sh -------------------------------------------------------------------------------- /PDE-Solving-StandardBenchmark/scripts/Transolver_Plas.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/PDE-Solving-StandardBenchmark/scripts/Transolver_Plas.sh -------------------------------------------------------------------------------- /PDE-Solving-StandardBenchmark/utils/normalizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/PDE-Solving-StandardBenchmark/utils/normalizer.py -------------------------------------------------------------------------------- /PDE-Solving-StandardBenchmark/utils/testloss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/PDE-Solving-StandardBenchmark/utils/testloss.py -------------------------------------------------------------------------------- /Physics_Attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/Physics_Attention.py -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/README.md -------------------------------------------------------------------------------- /pic/Transolver.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/pic/Transolver.png -------------------------------------------------------------------------------- /pic/physical_states.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/pic/physical_states.png -------------------------------------------------------------------------------- /pic/showcases.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Transolver/HEAD/pic/showcases.png --------------------------------------------------------------------------------