diff options
Diffstat (limited to 'training/run.py')
| -rw-r--r-- | training/run.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/training/run.py b/training/run.py index 06426e0..0e17b4e 100644 --- a/training/run.py +++ b/training/run.py @@ -14,7 +14,6 @@ from pytorch_lightning import ( from pytorch_lightning.loggers import LightningLoggerBase from torch import nn from torchinfo import summary -from text_recognizer.data.base_mapping import AbstractMapping import utils @@ -40,6 +39,7 @@ def run(config: DictConfig) -> Optional[float]: model: LightningModule = hydra.utils.instantiate( config.model, network=network, + mapping=datamodule.mapping, loss_fn=loss_fn, optimizer_configs=config.optimizers, lr_scheduler_configs=config.lr_schedulers, |