search

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
947 stars 222 forks source link

CorrDiff: Full inference verification with the Modulus checkpoints #407

Closed mnabian closed 6 months ago

mnabian commented 6 months ago

Checkpoints are published here: https://catalog.ngc.nvidia.com/orgs/nvidia/teams/modulus/models/corrdiff_inference_package

Comparison:

Regression MAE: Radar: paper: 0.77, modulus: 0.77 u10m: paper: 0.45, modulus: 0.45 v10m: paper: 0.50, modulus: 0.50 t2m: paper: 0.24, modulus: 0.24

Regression + Diffusion CRPS: Radar: paper: 0.53, modulus: 0.53 u10m: paper: 0.29, modulus: 0.29 v10m: paper: 0.31, modulus: 0.31 t2m: paper: 0.14, modulus: 0.14