diff --git a/hydra_flash/train.py b/hydra_flash/train.py index 90fc12415313ba54abf2653d890b0f26d5ddeaad..b384567e0823b1152c624d1b0303c394f2466b1b 100644 --- a/hydra_flash/train.py +++ b/hydra_flash/train.py @@ -15,7 +15,7 @@ instantiate = partial(hydra_instantiate, _convert_="all") def main(cfg): # Preprocess : download and/or unzip data call(cfg.preprocess) - print(OmegaConf.to_container(cfg.datamodule, resolve=True)) + datamodule = instantiate(cfg.datamodule) model_kwargs = {}