diff options
| author | aktersnurra <gustaf.rydholm@gmail.com> | 2020-11-12 23:42:03 +0100 |
|---|---|---|
| committer | aktersnurra <gustaf.rydholm@gmail.com> | 2020-11-12 23:42:03 +0100 |
| commit | 8fdb6435e15703fa5b76df19728d905650ee1aef (patch) | |
| tree | be3bec9e5cab4ef7f9d94528d102e57ce9b16c3f /src/text_recognizer/networks/wide_resnet.py | |
| parent | dc28cbe2b4ed77be92ee8b2b69a20689c3bf02a4 (diff) | |
| parent | 6cb08a110620ee09fe9d8a5d008197a801d025df (diff) | |
Working cnn transformer.
Diffstat (limited to 'src/text_recognizer/networks/wide_resnet.py')
| -rw-r--r-- | src/text_recognizer/networks/wide_resnet.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/text_recognizer/networks/wide_resnet.py b/src/text_recognizer/networks/wide_resnet.py index aa79c12..28f3380 100644 --- a/src/text_recognizer/networks/wide_resnet.py +++ b/src/text_recognizer/networks/wide_resnet.py @@ -2,7 +2,7 @@ from functools import partial from typing import Callable, Dict, List, Optional, Type, Union -from einops.layers.torch import Rearrange, Reduce +from einops.layers.torch import Reduce import numpy as np import torch from torch import nn |