diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-10-10 18:06:20 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-10-10 18:06:20 +0200 |
| commit | cba94bdcab90f288dd1172607500ba2b28279736 (patch) | |
| tree | 3669f323171838ed17502978fae23695ae3591cf | |
| parent | 9d3e158f48e3e965d67f154e05faaaf1cadbb836 (diff) | |
Update initialization
| -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, |