summaryrefslogtreecommitdiff
path: root/text_recognizer/model/base.py
diff options
context:
space:
mode:
Diffstat (limited to 'text_recognizer/model/base.py')
-rw-r--r--text_recognizer/model/base.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/text_recognizer/model/base.py b/text_recognizer/model/base.py
index 9a751bf..c662583 100644
--- a/text_recognizer/model/base.py
+++ b/text_recognizer/model/base.py
@@ -2,11 +2,11 @@
from typing import Any, Dict, Optional, Tuple, Type
import hydra
+import pytorch_lightning as L
import torch
from loguru import logger as log
from omegaconf import DictConfig
-import pytorch_lightning as L
-from torch import nn, Tensor
+from torch import Tensor, nn
from text_recognizer.data.tokenizer import Tokenizer