summaryrefslogtreecommitdiff
path: root/text_recognizer/model/base.py
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2024-04-15 21:47:54 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2024-04-15 21:47:54 +0200
commit9ce21f569ecac03f15f2ad669fde3dd4a512f8cc (patch)
treee6f5bbf4cfe758788fd6ad3679b714d4ecfad568 /text_recognizer/model/base.py
parenta93dcc5b9c8160a441c5b5f99f2f59264778ef91 (diff)
Format
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