Skip to content
This repository has been archived by the owner on Mar 21, 2024. It is now read-only.

Commit

Permalink
comments
Browse files Browse the repository at this point in the history
  • Loading branch information
ant0nsc committed Jul 27, 2021
1 parent de1ace2 commit 0b744da
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion InnerEye/ML/model_training.py
Original file line number Diff line number Diff line change
Expand Up @@ -130,7 +130,7 @@ def create_lightning_trainer(container: LightningContainer,
if effective_num_gpus > 1:
accelerator: Optional[str] = "ddp"
# Initialize the DDP plugin. The default for pl_find_unused_parameters is False. If True, the plugin prints out
# lengthy warnings about the performance impact of find_unused_parameters
# lengthy warnings about the performance impact of find_unused_parameters.
plugins = [DDPPlugin(num_nodes=num_nodes, sync_batchnorm=True,
find_unused_parameters=container.pl_find_unused_parameters)]
else:
Expand Down

0 comments on commit 0b744da

Please # to comment.