NVIDIA / modulus

Open-source deep-learning framework for building, training, and fine-tuning deep learning models using state-of-the-art Physics-ML methods
https://developer.nvidia.com/modulus
Apache License 2.0
937 stars 219 forks source link

AeroGraphNet example #544

Closed Alexey-Kamenev closed 3 months ago

Alexey-Kamenev commented 3 months ago

Modulus Pull Request

Description

This example demonstrates how to train MeshGraphNet model on two automotive aerodynamics datasets, Ahmed Body and DrivAerNet.

The example uses Hydra experiments as a way to specify experiment configuration. For example, to train default MGN on DrivAerNet dataset, the following command can be used:

python train.py +experiment=drivaernet/mgn data.data_dir=/data/DrivAerNet/

See the example README for more information.

The example also contains visualization of the meshes which allows to better interpret the training progress.

Checklist

Dependencies

Alexey-Kamenev commented 3 months ago

/blossom-ci

Alexey-Kamenev commented 3 months ago

/blossom-ci

Alexey-Kamenev commented 3 months ago

/blossom-ci

mnabian commented 3 months ago

This PR looks great! Left a few minor comments.

Alexey-Kamenev commented 3 months ago

/blossom-ci

Alexey-Kamenev commented 3 months ago

/blossom-ci