diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2023-08-25 23:19:14 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2023-08-25 23:19:14 +0200 |
| commit | 49ca6ade1a19f7f9c702171537fe4be0dfcda66d (patch) | |
| tree | 20062ed1910758481f3d5fff11159706c7b990c6 /text_recognizer/data | |
| parent | 0421daf6bd97596703f426ba61c401599b538eeb (diff) | |
Rename and add flash atten
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__( |