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
937 stars 219 forks source link

Update logic on corrdiff patch based training #524

Closed daviddpruitt closed 4 months ago

daviddpruitt commented 4 months ago

The existing patch based training check automatically assumes that the patch size is not the full region. If a full domain region that is rectangular is being trained It is incorrectly rejected as being patch based.

Modulus Pull Request

Description

Updates the logic on the patch based size checking to avoid incorrectly rejecting full domain trainings. Fixes issue: https://github.com/NVIDIA/modulus/issues/522

Checklist

Dependencies

daviddpruitt commented 4 months ago

/blossom-ci