NVIDIA / modulus

Open-source deep-learning framework for building, training, and fine-tuning deep learning models using state-of-the-art Physics-ML methods

Apache License 2.0

791 stars 165 forks source link

Modulus Pull Request

Description

Add rectangular image support to full domain and patch-based training and inference. Support for 2d image sizes is added to SongUNet, Unet, and ResLoss. If a single dimension is given a square region is assumed retaining the original functionality.

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.
[ ] An issue is linked to this pull request.

NVIDIA / modulus

[Enh] [CorrDiff] Support rectangular regions in CorrDiff #540

Modulus Pull Request

Description

Checklist

Dependencies