From 8f61ff92feeb798cc9e0f07fae48d412dd99a65e Mon Sep 17 00:00:00 2001 From: Yufei Xu Date: Tue, 24 May 2022 13:14:28 +0800 Subject: [PATCH] Update train.py change the default manner of find_unused_parameters --- mmpose/apis/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mmpose/apis/train.py b/mmpose/apis/train.py index 2fd1771..7c31f8b 100644 --- a/mmpose/apis/train.py +++ b/mmpose/apis/train.py @@ -114,7 +114,7 @@ def train_model(model, # put model on gpus if distributed: - find_unused_parameters = cfg.get('find_unused_parameters', True) + find_unused_parameters = cfg.get('find_unused_parameters', False) # Sets the `find_unused_parameters` parameter in # torch.nn.parallel.DistributedDataParallel