From 3b06ef615a8db67a03927576e0c12fbfb2501f5f Mon Sep 17 00:00:00 2001 From: aktersnurra Date: Mon, 14 Sep 2020 22:15:47 +0200 Subject: Fixed CTC loss. --- src/text_recognizer/networks/ctc.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'src/text_recognizer/networks/ctc.py') diff --git a/src/text_recognizer/networks/ctc.py b/src/text_recognizer/networks/ctc.py index 72f18b8..2493d5c 100644 --- a/src/text_recognizer/networks/ctc.py +++ b/src/text_recognizer/networks/ctc.py @@ -24,7 +24,7 @@ def greedy_decoder( target_lengths (Optional[Tensor]): Length of each target tensor. Defaults to None. character_mapper (Optional[Callable]): A emnist/character mapper for mapping integers to characters. Defaults to None. - blank_label (int): The blank character to be ignored. Defaults to 79. + blank_label (int): The blank character to be ignored. Defaults to 80. collapse_repeated (bool): Collapase consecutive predictions of the same character. Defaults to True. Returns: -- cgit v1.2.3-70-g09d2