UNet 3D: Support gradient checkpointing

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

798 stars 174 forks source link

Open mnabian opened 2 months ago