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
797 stars 174 forks source link

Enable CorrDiff multi-gpu generation #472

Closed tge25 closed 2 months ago

tge25 commented 2 months ago

Modulus Pull Request

Description

Fix multi-gpu generation bugs Closes #348

Checklist

Dependencies

mnabian commented 2 months ago

/blossom-ci