Browse Source
Update train.py
change the default manner of find_unused_parameters
233
Yufei Xu
3 years ago
committed by
GitHub
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with
1 additions and
1 deletions
-
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 |
|
|
|
|
|
|
|
|
