diff options
| -rw-r--r-- | training/run.py | 8 |
1 files changed, 1 insertions, 7 deletions
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, |