diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-05-02 13:51:15 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-05-02 13:51:15 +0200 |
| commit | 1d0977585f01c42e9f6280559a1a98037907a62e (patch) | |
| tree | 7e86dd71b163f3138ed2658cb52c44e805f21539 /text_recognizer/networks/transformer/__init__.py | |
| parent | 58ae7154aa945cfe5a46592cc1dfb28f0a4e51b3 (diff) | |
Implemented training script with hydra
Diffstat (limited to 'text_recognizer/networks/transformer/__init__.py')
| -rw-r--r-- | text_recognizer/networks/transformer/__init__.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/text_recognizer/networks/transformer/__init__.py b/text_recognizer/networks/transformer/__init__.py index 627fa7b..4ff48f7 100644 --- a/text_recognizer/networks/transformer/__init__.py +++ b/text_recognizer/networks/transformer/__init__.py @@ -4,4 +4,5 @@ from .positional_encoding import ( PositionalEncoding2D, target_padding_mask, ) -from .transformer import Decoder, DecoderLayer, Encoder, EncoderLayer, Transformer + +# from .transformer import Decoder, DecoderLayer, Encoder, EncoderLayer, Transformer |