search

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
784 stars 164 forks source link

UNet: Add support for 2D #512

Open mnabian opened 1 month ago

DavidLandup0 commented 1 month ago

@mnabian I could take this off your plate if there are higher priority items in the UNet epic :)