summaryrefslogtreecommitdiff
path: root/src/text_recognizer/networks
diff options
context:
space:
mode:
authoraktersnurra <gustaf.rydholm@gmail.com>2020-08-20 22:56:29 +0200
committeraktersnurra <gustaf.rydholm@gmail.com>2020-08-20 22:56:29 +0200
commitfe23001b6588e6e6e9e2c5a99b72f3445cf5206f (patch)
treea12dffe669ee2b53bc3be2058b3adfaae03ea611 /src/text_recognizer/networks
parent1f459ba19422593de325983040e176f97cf4ffc0 (diff)
Minor bug fix.
Diffstat (limited to 'src/text_recognizer/networks')
-rw-r--r--src/text_recognizer/networks/line_lstm_ctc.py1
1 files changed, 1 insertions, 0 deletions
diff --git a/src/text_recognizer/networks/line_lstm_ctc.py b/src/text_recognizer/networks/line_lstm_ctc.py
index d704139..2e2c3a5 100644
--- a/src/text_recognizer/networks/line_lstm_ctc.py
+++ b/src/text_recognizer/networks/line_lstm_ctc.py
@@ -2,3 +2,4 @@
import torch
from torch import nn
+from torch import Tensor