From 36291af8adc75907c11d011546d59b0bf0a5a65d Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Sat, 28 Jan 2023 02:50:25 +0900 Subject: [PATCH] from local --- .../.train.py.swp | Bin 53248 -> 53248 bytes .../ASR/pruned_transducer_stateless5/train.py | 2 +- 2 files changed, 1 insertion(+), 1 deletion(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless5/.train.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless5/.train.py.swp index 5453785f363a172549b81265624a67f385afae55..6160286772ae6125ccf7a3c3e95816af2fa8d341 100644 GIT binary patch delta 219 zcmWN_Jqtl`7>4nOAL*SZu>b%7 delta 218 zcmWN_Jqtl`7>4nOzjH8%$zhSnqNG?X-WC}UNo-=UE3=n{tPE_vf=)6TDG8YrBOAVg z>#kqjU4xYxtd!A;*7Aj5A`%tm5!t#o?aR`!Ha;mWLefoVI%etH?OYsmkv&Ejq6j~p zn#cn;?65!wO@#5YiF|Rw5@R&rgMqgadEtmTrl_EdkP@%lts-YkQAZK_?3#(s0wbsC P$|f~c?e@%4Zaw(}L(M26 diff --git a/egs/librispeech/ASR/pruned_transducer_stateless5/train.py b/egs/librispeech/ASR/pruned_transducer_stateless5/train.py index 378e44612..ff70348e9 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless5/train.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless5/train.py @@ -983,7 +983,7 @@ def run(rank, world_size, args): model.to(device) if world_size > 1: logging.info("Using DDP") - model = DDP(model, device_ids=[rank], find_unused_parameters=True) + model = DDP(model, device_ids=[rank], find_unused_parameters=False) optimizer = Eve(model.parameters(), lr=params.initial_lr)