From cba94bdcab90f288dd1172607500ba2b28279736 Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Sun, 10 Oct 2021 18:06:20 +0200 Subject: Update initialization --- training/run.py | 8 +------- 1 file changed, 1 insertion(+), 7 deletions(-) (limited to 'training/run.py') diff --git a/training/run.py b/training/run.py index b854376..06426e0 100644 --- a/training/run.py +++ b/training/run.py @@ -27,13 +27,8 @@ def run(config: DictConfig) -> Optional[float]: if config.get("seed"): seed_everything(config.seed, workers=True) - log.info(f"Instantiating mapping <{config.mapping._target_}>") - mapping: AbstractMapping = hydra.utils.instantiate(config.mapping) - log.info(f"Instantiating datamodule <{config.datamodule._target_}>") - datamodule: LightningDataModule = hydra.utils.instantiate( - config.datamodule, mapping=mapping - ) + datamodule: LightningDataModule = hydra.utils.instantiate(config.datamodule) log.info(f"Instantiating network <{config.network._target_}>") network: nn.Module = hydra.utils.instantiate(config.network) @@ -44,7 +39,6 @@ def run(config: DictConfig) -> Optional[float]: log.info(f"Instantiating model <{config.model._target_}>") model: LightningModule = hydra.utils.instantiate( config.model, - mapping=mapping, network=network, loss_fn=loss_fn, optimizer_configs=config.optimizers, -- cgit v1.2.3-70-g09d2