NVIDIA / earth2mip

Earth-2 Model Intercomparison Project (MIP) is a python framework that enables climate researchers and scientists to inter-compare AI models for weather and climate.

https://nvidia.github.io/earth2mip/

Apache License 2.0

187 stars 41 forks source link

use device correctly in time_collection #100

Closed nbren12 closed 10 months ago

nbren12 commented 10 months ago

Earth-2 MIP Pull Request

Description

All models were loaded on the default device in parallel jobs. not good.

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

Dependencies

nbren12 commented 10 months ago

/blossom-ci