-- added embedding dim to categorymdn

Author: manujosephv

pytorch_tabular/models/mixture_density/mdn.py

@@ -253,6 +253,7 @@ def validation_epoch_end(self, outputs) -> None:
 
 class CategoryEmbeddingMDN(BaseMDN):
     def __init__(self, config: DictConfig, **kwargs):
+        self.embedding_cat_dim = sum([y for x, y in config.embedding_dims])
         super().__init__(config, **kwargs)
 
     def _build_network(self):