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
803 stars 178 forks source link

[Enhancement] Add Inference Script to CorrDiff Example #384

Open dallasfoster opened 4 months ago

dallasfoster commented 4 months ago

It would be beneficial to have a simplified script in the CorrDiff example that shows how to run inference on a single netcdf/zarr/xarray array. This inference script would be minimal but would have necessary IO to be run as CLI. It might also be beneficial to automatically download data from ARCO ERA5 for users.

dallasfoster commented 4 months ago

The availability of a checkpoint is a concern.