Skip to content

Commit 99cece8

Browse files
committed
Fix message
1 parent 126aedb commit 99cece8

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

pytorch_lightning/plugins/training_type/ddp.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -258,8 +258,8 @@ def pre_configure_ddp(self):
258258
):
259259
# TODO: PyTorch 1.7.0 DDP introduces `self.reducer._rebuild_buckets()` breaking manual_optimization
260260
rank_zero_warn(
261-
"Lightning `manual_optimization` needs to set `find_unused_parameters=True` to properly work with DDP."
262-
" Using `find_unused_parameters=True`."
261+
"From PyTorch 1.7.0, Lightning `manual_optimization` needs to set `find_unused_parameters=True` to"
262+
" properly work with DDP. Using `find_unused_parameters=True`."
263263
)
264264
self._ddp_kwargs["find_unused_parameters"] = True
265265

pytorch_lightning/plugins/training_type/ddp_spawn.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -242,8 +242,8 @@ def pre_configure_ddp(self):
242242
):
243243
# TODO: PyTorch 1.7.0 DDP introduces `self.reducer._rebuild_buckets()` breaking manual_optimization
244244
rank_zero_warn(
245-
"Lightning `manual_optimization` needs to set `find_unused_parameters=True` to properly work with DDP."
246-
" Using `find_unused_parameters=True`."
245+
"From PyTorch 1.7.0, Lightning `manual_optimization` needs to set `find_unused_parameters=True` to"
246+
" properly work with DDP. Using `find_unused_parameters=True`."
247247
)
248248
self._ddp_kwargs["find_unused_parameters"] = True
249249

0 commit comments

Comments
 (0)