From e5a947bc8a916bd762047f46f942c6ace49638ac Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Sat, 28 Jan 2023 02:54:10 +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 6160286772ae6125ccf7a3c3e95816af2fa8d341..88e21eee9b178ca7432a6ea565a76fb2fcddff18 100644 GIT binary patch delta 213 zcmWN_KMMhI9LMo5KX;dsEDoE=qNGep`LoF2L=u}A?8@v|$jZRxDd--6k&+}9MOhfn zQ}F)Ot3JK$t+Kbu?4?_kYCM}t3k!+7gZF+ovNPShmZIp6zmV~a4TFoLE^@#W6Ih7h zrHMRo!yao4&_xnIpNNYyHkhGy1tGmnI81}}{1G8PX61j{iEGjaN+=myv)xV-jQJr|WRE#)Scv1L zM4q@|hZRQXA%$N+_*ZM*yfEN&)O diff --git a/egs/librispeech/ASR/pruned_transducer_stateless5/train.py b/egs/librispeech/ASR/pruned_transducer_stateless5/train.py index ff70348e9..378e44612 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=False) + model = DDP(model, device_ids=[rank], find_unused_parameters=True) optimizer = Eve(model.parameters(), lr=params.initial_lr)