From fe23001b6588e6e6e9e2c5a99b72f3445cf5206f Mon Sep 17 00:00:00 2001
From: aktersnurra <gustaf.rydholm@gmail.com>
Date: Thu, 20 Aug 2020 22:56:29 +0200
Subject: Minor bug fix.

---
 src/text_recognizer/models/base.py                 |   4 ++--
 src/text_recognizer/networks/line_lstm_ctc.py      |   1 +
 ...rModel_EmnistDataset_ResidualNetwork_weights.pt | Bin 28654593 -> 4562821 bytes
 3 files changed, 3 insertions(+), 2 deletions(-)

(limited to 'src/text_recognizer')

diff --git a/src/text_recognizer/models/base.py b/src/text_recognizer/models/base.py
index 74fd223..3a84a11 100644
--- a/src/text_recognizer/models/base.py
+++ b/src/text_recognizer/models/base.py
@@ -139,10 +139,10 @@ class Model(ABC):
         else:
             _optimizer = None
 
-        if self._optimizer and lr_scheduler is not None:
+        if _optimizer and lr_scheduler is not None:
             if "OneCycleLR" in str(lr_scheduler):
                 lr_scheduler_args["steps_per_epoch"] = len(self._data_loaders["train"])
-            _lr_scheduler = lr_scheduler(self._optimizer, **lr_scheduler_args)
+            _lr_scheduler = lr_scheduler(_optimizer, **lr_scheduler_args)
         else:
             _lr_scheduler = None
 
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
diff --git a/src/text_recognizer/weights/CharacterModel_EmnistDataset_ResidualNetwork_weights.pt b/src/text_recognizer/weights/CharacterModel_EmnistDataset_ResidualNetwork_weights.pt
index 008beb2..a5c6aaf 100644
Binary files a/src/text_recognizer/weights/CharacterModel_EmnistDataset_ResidualNetwork_weights.pt and b/src/text_recognizer/weights/CharacterModel_EmnistDataset_ResidualNetwork_weights.pt differ
-- 
cgit v1.2.3-70-g09d2