Fix datamodule config

grok-ai · Sep 13, 2023 · 1eb9677 · 1eb9677
1 parent 8471d9a
commit 1eb9677
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__":