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

791 stars 165 forks source link

[Fix] [CorrDiff] Diffusion fixes #567

Open daviddpruitt opened 5 days ago

daviddpruitt commented 5 days ago

Modulus Pull Request

Description

Fixes bugs in Diffusion training scripts and layers.

Resolves https://github.com/NVIDIA/modulus/issues/541

Checklist

[x] I am familiar with the Contributing Guidelines.
[x] New or existing tests cover these changes.
[x] The documentation is up to date with these changes.
[x] The CHANGELOG.md is up to date with these changes.
[x] An issue is linked to this pull request.

Dependencies