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

941 stars 220 forks source link

Address intermittent CI failures #480

Closed ktangsali closed 5 months ago

ktangsali commented 5 months ago

Modulus Pull Request

Description

CI failed intermittently on the metrics/test_loss_climate.py, while the exact cause is unknown, reducing the size of the tensor seems to help with the intermittency.

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.
[ ] The CHANGELOG.md is up to date with these changes.
[ ] An issue is linked to this pull request.

Dependencies

ktangsali commented 5 months ago

/blossom-ci

daviddpruitt commented 5 months ago

Looks good to me