NVIDIA / earth2studio

Open-source deep-learning framework for exploring, building and deploying AI weather/climate workflows.

https://nvidia.github.io/earth2studio/

Apache License 2.0

95 stars 31 forks source link

Skip interpolation in map_coords when possible #109

Closed jleinonen closed 3 months ago

jleinonen commented 3 months ago

Earth2Studio Pull Request

Description

Optimizes map_coords to skip interpolation whenever the output coordinates are identical to or a continuous subset of the input coordinates.

Tested to work with SFNO and FCN models.

Checklist

[x] I am familiar with the Contributing Guidelines.
[x] New or existing tests cover these changes.
[ ] 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

jleinonen commented 3 months ago

/blossom-ci

NickGeneva commented 3 months ago

/blossom-ci

NickGeneva commented 3 months ago

Closing, updates merged in linked PR. Thanks!