External Aerodynamics (Python Scripts)

Note

See the Walkthrough: Aerodynamic CFD with Noether for the full tutorial.

Source code: recipes/aero_cfd/scripts/

Available training scripts

Script

Dataset

Source

train_ahmedml.py

AhmedML (CAEML benchmark)

CAEML

train_drivaerml.py

DrivAerML (CAEML benchmark)

CAEML

train_drivaernet.py

DrivAerNet++

DrivAerNet

train_emmi_wing.py

Emmi Wing

EmmiAI /Emmi Wing

train_shapenet_car.py

ShapeNet Car

ShapeNet

Each script contains functions for training different model architectures (AB-UPT, UPT, Transformer, Transolver).

How to run

From the recipes directory run:

uv run python -m aero_cfd.scripts.train_shapenet_car \
  --dataset-root /path/to/shapenet_car \
  --output-path /path/to/outputs \
  --accelerator gpu \
  --model abupt

Available arguments:

  • --dataset-root (required) — Path to the dataset.

  • --output-path (required) — Path to store training outputs.

  • --accelerator — Accelerator to use: cpu, gpu, or mps (default: gpu).

  • --model — Model architecture: abupt, upt, transformer, or transolver (default: abupt).