CorrDiff model for Continental US

[gowtham-talluru]

Hi Team

Is trained CorrDiff model available for continental US?

Thanks
Gowtham

[NickGeneva]

Hi @gowtham-talluru

Not that is a part of Earth2Studio right as this moment. There is a checkpoint that us provided as part the the NIM developer program / Nvidia AIE but its not integrated with the package yet.

The corrdiff code base has had a lot of improvements, so we are waiting for that to be fully complete before any assessments of adding a support for it in E2S with the improved inference code. PhysicsNeMo has the recipes for training. Thanks!

[ram-cherukuri]

You can find the pretrained model for CONUS as NIM here: https://catalog.ngc.nvidia.com/orgs/nim/teams/nvidia/containers/corrdiff?version=1.0.0
The documentation is here: https://docs.nvidia.com/nim/earth-2/corrdiff/latest/overview.html