From 7279f1cc188ff38233a5f558759eef04603b0e67 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Sol=C3=A8ne=20Tarride?= <starride@teklia.com>
Date: Thu, 12 Oct 2023 15:36:06 +0200
Subject: [PATCH] Fix docstring again
---
dan/ocr/decoder.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/dan/ocr/decoder.py b/dan/ocr/decoder.py
index bba35b99..496d9b2e 100644
--- a/dan/ocr/decoder.py
+++ b/dan/ocr/decoder.py
@@ -476,7 +476,7 @@ class CTCLanguageDecoder:
:param tokens_path: Path to a file containing valid tokens. If using a file, the expected
format is for tokens mapping to the same index to be on the same line.
:param language_model_weight: Weight of the language model.
- :param temperature_scaling: Temperature for model calibreation.
+ :param temperature: Temperature for model calibreation.
"""
def __init__(
@@ -495,7 +495,7 @@ class CTCLanguageDecoder:
}
self.index_to_token = {i: token for token, i in self.tokens_to_index.items()}
self.blank_token_id = self.tokens_to_index[self.mapping.ctc.encoded]
-
+
# Torchaudio's decoder
# https://pytorch.org/audio/master/generated/torchaudio.models.decoder.ctc_decoder.html
self.decoder = ctc_decoder(
--
GitLab