Skip to content

Commit f11deb2

Browse files
committed
Addressing #56
Signed-off-by: Oleksii Kuchaiev <[email protected]>
1 parent 891681e commit f11deb2

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

docs/sources/source/asr/tutorial.rst

+2
Original file line numberDiff line numberDiff line change
@@ -290,6 +290,8 @@ Training time can be dramatically reduced if starting from a good pre-trained mo
290290
291291
jasper_encoder.restore_from("<path_to_checkpoints>/15x5SEP/JasperEncoder-STEP-247400.pt")
292292
jasper_decoder.restore_from("<path_to_checkpoints>/15x5SEP/JasperDecoderForCTC-STEP-247400.pt")
293+
# in case of distributed training add args.local_rank
294+
jasper_decoder.restore_from("<path_to_checkpoints>/15x5SEP/JasperDecoderForCTC-STEP-247400.pt", args.local_rank)
293295
294296
.. tip::
295297
When fine-tuning, use smaller learning rate.

0 commit comments

Comments
 (0)