Merge pull request #95 from LeonardoEmili/fix-datamodule-config

Fix datamodule config
grok-ai · Oct 10, 2023 · 306b461 · 306b461
2 parents 6be9db1 + 1eb9677
commit 306b461
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/{{ cookiecutter.repository_name }}/src/{{ cookiecutter.package_name }}/data/datamodule.py b/{{ cookiecutter.repository_name }}/src/{{ cookiecutter.package_name }}/data/datamodule.py
@@ -214,7 +214,7 @@ def main(cfg: omegaconf.DictConfig) -> None:
     Args:
         cfg: the hydra configuration
     """
-    _: pl.LightningDataModule = hydra.utils.instantiate(cfg.data.datamodule, _recursive_=False)
+    _: pl.LightningDataModule = hydra.utils.instantiate(cfg.nn.data, _recursive_=False)
 
 
 if __name__ == "__main__":