From 8e5e6cc357d6a7d699adf47009f53f49a6df05e8 Mon Sep 17 00:00:00 2001 From: Michael Fuest Date: Sun, 29 Dec 2024 15:29:07 -0500 Subject: [PATCH] diffcharge update --- generator/diffcharge/diffusion.py | 1 + 1 file changed, 1 insertion(+) diff --git a/generator/diffcharge/diffusion.py b/generator/diffcharge/diffusion.py index f327f17..c1e9474 100644 --- a/generator/diffcharge/diffusion.py +++ b/generator/diffcharge/diffusion.py @@ -94,6 +94,7 @@ def __init__(self, cfg: DictConfig): self.beta_start = cfg.model.beta_start self.beta_end = cfg.model.beta_end self.current_epoch = 0 + self.conditioning_var_n_categories = cfg.dataset.conditioning_vars self.conditioning_module = ConditioningModule( categorical_dims=cfg.dataset.conditioning_vars,