modulus
modulus copied to clipboard
NVIDIA
Open-source deep-learning framework for building, training, and fine-tuning deep learning models using state-of-the-art Physics-ML methods
```[tasklist] ### Tasks - [ ] https://github.com/NVIDIA/modulus/issues/464 - [ ] https://github.com/NVIDIA/modulus/issues/465 - [ ] https://github.com/NVIDIA/modulus/issues/466 - [ ] https://github.com/NVIDIA/modulus/issues/467 - [ ] UNet 3D: Support spatial parallelism ```
### Version Latest ### On which installation method(s) does this occur? _No response_ ### Describe the issue Traceback (most recent call last): File "/code/mnabian/PRs/out/modulus/examples/generative/corrdiff/train.py", line 341, in main training_loop.training_loop( File...
### Is this a new feature, an improvement, or a change to existing functionality? Improvement ### How would you describe the priority of this feature request Low (would be nice)...
### Version main ### On which installation method(s) does this occur? Source ### Describe the issue Many times, local development requires building custom docker containers with different dependencies or packages....
### Is this a new feature, an improvement, or a change to existing functionality? New Feature ### How would you describe the priority of this feature request Critical (currently preventing...
**not for this PR since this code already exists** Eventually, the loss should not know anything about patches and just treat them like batches (hey this rhymes). This can be...
