diff options
| -rw-r--r-- | text_recognizer/models/__init__.py | 1 | ||||
| -rw-r--r-- | text_recognizer/networks/__init__.py | 1 |
2 files changed, 2 insertions, 0 deletions
diff --git a/text_recognizer/models/__init__.py b/text_recognizer/models/__init__.py index 1982daf..cc02487 100644 --- a/text_recognizer/models/__init__.py +++ b/text_recognizer/models/__init__.py @@ -1 +1,2 @@ """PyTorch Lightning models modules.""" +from text_recognizer.models.transformer import LitTransformer diff --git a/text_recognizer/networks/__init__.py b/text_recognizer/networks/__init__.py index f921882..dad9d23 100644 --- a/text_recognizer/networks/__init__.py +++ b/text_recognizer/networks/__init__.py @@ -1,2 +1,3 @@ """Network modules""" from text_recognizer.networks.conv_transformer import ConvTransformer +from text_recognizer.networks.efficientnet.efficientnet import EfficientNet |