diff options
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 |