diff options
Diffstat (limited to 'text_recognizer/data')
| -rw-r--r-- | text_recognizer/data/base_data_module.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/text_recognizer/data/base_data_module.py b/text_recognizer/data/base_data_module.py index bd6fd99..3cb0aa2 100644 --- a/text_recognizer/data/base_data_module.py +++ b/text_recognizer/data/base_data_module.py @@ -2,7 +2,7 @@ from pathlib import Path from typing import Callable, Dict, Optional, Tuple, TypeVar -from pytorch_lightning import LightningDataModule +import pytorch_lightning as L from torch.utils.data import DataLoader from text_recognizer.data.base_dataset import BaseDataset @@ -19,7 +19,7 @@ def load_and_print_info(data_module_class: type) -> None: print(dataset) -class BaseDataModule(LightningDataModule): +class BaseDataModule(L.LightningDataModule): """Base PyTorch Lightning DataModule.""" def __init__( |