From e193ca9d94456c1933e25d2f8b7e8224d3e92ae3 Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Fri, 10 Jun 2022 00:33:46 +0200 Subject: Add imports --- text_recognizer/models/__init__.py | 1 + text_recognizer/networks/__init__.py | 1 + 2 files changed, 2 insertions(+) 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 -- cgit v1.2.3-70-g09d2