From ba337f8554c2b0b7e0ab3462027de59862cb95dc Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Mon, 29 May 2023 11:47:03 +0800 Subject: [PATCH] find_unused_parameters=True removed --- egs/libriheavy/LM/zipformer1/train.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/egs/libriheavy/LM/zipformer1/train.py b/egs/libriheavy/LM/zipformer1/train.py index 903586e8e..fddf6e59d 100755 --- a/egs/libriheavy/LM/zipformer1/train.py +++ b/egs/libriheavy/LM/zipformer1/train.py @@ -953,8 +953,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]) optimizer = ScaledAdam( get_parameter_groups_with_lrs(