Modelpublic

Diffusion-Research-Lab/hrrr-precipitation-100x100-diffusion

sourceHugging Faceupdated 21d agoView on Hugging Face

0likes

Model Card

hrrr

DDPM-V and DLPM-Eps reference models trained with gendynamics. Each model subdirectory contains its architecture, training settings, normalization, metrics, and weights.

ddpm-v

—best_epoch: 70
—validation_loss: 0.073888
—test_loss: 0.0758872

dlpm-eps

—best_epoch: 122
—validation_loss: 0.153031
—test_loss: 0.179959

Loading

python

from toolkit.huggingface import load_huggingface_model

model, normalization, data_scale = load_huggingface_model("Diffusion-Research-Lab/hrrr-precipitation-100x100-diffusion", "ddpm_v", device="cuda")
samples = model.sample(16).float().cpu()
if normalization is not None:
    samples = samples * normalization["std"] + normalization["mean"]