From f3da6d8c9526ce4ddb935440f08ba053be5cb2fb Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Sat, 10 Dec 2022 15:15:28 +0900 Subject: [PATCH] from local --- .../.train.py.swp | Bin 110592 -> 110592 bytes .../train.py | 2 +- 2 files changed, 1 insertion(+), 1 deletion(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train.py.swp index 4a5312cc2d5152bd28527293361c0b7522ae1263..0091e48ceef6bc41b932dba4b8897261d528c103 100644 GIT binary patch delta 141 zcmZp8z}E19O)SYE%+puFQqO<^2m}}yUMWvWzR%kzwpEPr-F#MC9tMW9(>WI~+BbjxCe+!fjAI|y@A*Wh$VrT9f-dGwO;|^b0A}QHY*Cd@J#R3 YW)$2krKHEmnwy`JnzLPK9pgJj0HduQh5!Hn delta 133 zcmZp8z}E19O)SYE%+puFQqO<^2n2vgX-cv&SEJZgF~)cES-TnT)=4W#KgzI zzzD<)KzxFifnhxm&jI3AAkG6~FCcaSVr?K60%8Urehbuo7Ko37jN#cnaRFl!FI#SY ON@-5&cA<5Q?-&6NEE}%? diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py index abd2dbd0a..4f54202a2 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py @@ -106,7 +106,7 @@ def set_batch_count(model: Union[nn.Module, DDP], batch_count: float) -> None: for module in model.modules(): if hasattr(module, "batch_count"): module.batch_count = batch_count - model.module.encoder.num_updates = int(batch_count) + model.encoder.num_updates = int(batch_count) def add_rep_arguments(parser: argparse.ArgumentParser):