From 49ca6ade1a19f7f9c702171537fe4be0dfcda66d Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Fri, 25 Aug 2023 23:19:14 +0200 Subject: Rename and add flash atten --- text_recognizer/data/base_data_module.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'text_recognizer/data') 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__( -- cgit v1.2.3-70-g09d2