Skip to content

Commit bcd1b3c

Browse files
committed
update doc string
Signed-off-by: Ngoc Minh Tran <[email protected]>
1 parent c739a5e commit bcd1b3c

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

art/attacks/evasion/imperceptible_asr/imperceptible_asr_pytorch.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -145,7 +145,8 @@ def __init__(
145145
:param opt_level: Specify a pure or mixed precision optimization level. Used when use_amp is True. Accepted
146146
values are `O0`, `O1`, `O2`, and `O3`.
147147
:param loss_scale: Loss scaling. Used when use_amp is True. Default is 1.0 due to warp-ctc not supporting
148-
scaling of gradients.
148+
scaling of gradients. If passed as a string, must be a string representing a number,
149+
e.g., “1.0”, or the string “dynamic”.
149150
"""
150151
import torch # lgtm [py/repeated-import]
151152
from torch.autograd import Variable

art/estimators/speech_recognition/pytorch_deep_speech.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -94,7 +94,8 @@ def __init__(
9494
:param opt_level: Specify a pure or mixed precision optimization level. Used when use_amp is True. Accepted
9595
values are `O0`, `O1`, `O2`, and `O3`.
9696
:param loss_scale: Loss scaling. Used when use_amp is True. Default is 1.0 due to warp-ctc not supporting
97-
scaling of gradients.
97+
scaling of gradients. If passed as a string, must be a string representing a number,
98+
e.g., “1.0”, or the string “dynamic”.
9899
:param decoder_type: Decoder type. Either `greedy` or `beam`. This parameter is only used when users want
99100
transcription outputs.
100101
:param lm_path: Path to an (optional) kenlm language model for use with beam search. This parameter is only

0 commit comments

Comments
 (0)