diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-06-10 00:33:46 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-06-10 00:33:46 +0200 |
| commit | e193ca9d94456c1933e25d2f8b7e8224d3e92ae3 (patch) | |
| tree | cdd982d3f4f13f9d019e87ca0ea12d99400baea1 | |
| parent | 7a8c9d40241868f3ecc5c8f5e4d0c06caa6c9e96 (diff) | |
Add imports
| -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 |