From 20d7904ece1a8bee8fa9ecf5766496cbbf6633e2 Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Mon, 11 Oct 2021 22:12:54 +0200 Subject: Add mapping to lit model --- training/run.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'training/run.py') 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, -- cgit v1.2.3-70-g09d2