Skip to content
Snippets Groups Projects
Commit 7279f1cc authored by Solene Tarride's avatar Solene Tarride
Browse files

Fix docstring again

parent d69453c8
No related branches found
No related tags found
1 merge request!274Add Language Model Decoder
This commit is part of merge request !274. Comments created here will be created in the context of that merge request.
......@@ -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(
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment