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

183 stars 40 forks source link

earth2mip/ensemble_utils.py #176

Closed ChenggongWang closed 6 months ago

ChenggongWang commented 6 months ago

Earth-2 MIP Pull Request

Description

Closes #175. Fixed the bug in perturbation_strategy: "spherical_grf". The bug is caused by missing device conversion when the ensemble inference is ran on GPU.