diff options
author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-07-30 23:15:03 +0200 |
---|---|---|
committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-07-30 23:15:03 +0200 |
commit | 7268035fb9e57342612a8cc50a1fe04e8841ca2f (patch) | |
tree | 8d4cf3743975bd25f2c04d6a56ff3d4608a7e8d9 /text_recognizer/networks/base.py | |
parent | 92fc1c7ed2f9f64552be8f71d9b8ab0d5a0a88d4 (diff) |
attr bug fix, properly loading network
Diffstat (limited to 'text_recognizer/networks/base.py')
-rw-r--r-- | text_recognizer/networks/base.py | 18 |
1 files changed, 0 insertions, 18 deletions
diff --git a/text_recognizer/networks/base.py b/text_recognizer/networks/base.py deleted file mode 100644 index 07b6a32..0000000 --- a/text_recognizer/networks/base.py +++ /dev/null @@ -1,18 +0,0 @@ -"""Base network with required methods.""" -from abc import abstractmethod - -import attr -from torch import nn, Tensor - - -@attr.s -class BaseNetwork(nn.Module): - """Base network.""" - - def __attrs_pre_init__(self) -> None: - super().__init__() - - @abstractmethod - def predict(self, x: Tensor) -> Tensor: - """Return token indices for predictions.""" - ... |