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
836 stars 190 forks source link

Add AMP support to CorrDiff #423

Closed jleinonen closed 3 months ago

jleinonen commented 3 months ago

Modulus Pull Request

Description

Add Automatic Mixed Precision (AMP) support to CorrDiff training:

Checklist

Dependencies

mnabian commented 3 months ago

/blossom-ci

mnabian commented 3 months ago

Closes https://github.com/NVIDIA/modulus/issues/424

jleinonen commented 3 months ago

/blossom-ci

mnabian commented 3 months ago

/blossom-ci

jleinonen commented 3 months ago

/blossom-ci

jleinonen commented 3 months ago

/blossom-ci

jleinonen commented 3 months ago

/blossom-ci

jleinonen commented 3 months ago

/blossom-ci

jleinonen commented 3 months ago

/blossom-ci