diff options
Diffstat (limited to 'text_recognizer/data')
| -rw-r--r-- | text_recognizer/data/transforms/pad.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/text_recognizer/data/transforms/pad.py b/text_recognizer/data/transforms/pad.py index 82e4d54..1da4534 100644 --- a/text_recognizer/data/transforms/pad.py +++ b/text_recognizer/data/transforms/pad.py @@ -14,7 +14,7 @@ class Pad: def __call__(self, y: Tensor) -> Tensor: """Pads sequences with pad index if shorter than max len.""" - if y.shape[-1] < self.length: + if y.shape[-1] < self.max_len: pad_len = self.max_len - len(y) y = torch.cat((y, torch.LongTensor([self.pad_index] * pad_len))) return y[: self.max_len] |