Commit aeb15613 authored by Binh Tang's avatar Binh Tang Committed by Facebook GitHub Bot
Browse files

Rename `DDPPlugin` to `DDPStrategy` (#11142)

Summary:
### New commit log messages
  b64dea9dc Rename `DDPPlugin` to `DDPStrategy` (#11142)

Reviewed By: jjenniferdai

Differential Revision: D33259306

fbshipit-source-id: b4608c6b96b4a7977eaa4ed3f03c4b824882aef0
parent 21ae9538
......@@ -25,10 +25,10 @@ from pytorch_lightning.loggers import TensorBoardLogger
from torch.distributed import get_rank
try:
from pytorch_lightning.plugins import DDPPlugin
from pytorch_lightning.plugins import DDPStrategy
except ImportError:
assert os.getenv("OSSRUN") == "1"
# FIXME: DDPPlugin has been renamed to DDPStrategy, however internal version is
# FIXME: DDPStrategy has been renamed to DDPStrategy, however internal version is
# not updated yet, temporally skipping the import in oss env in order to unblock
# CI where DPP is not used.
......@@ -90,7 +90,7 @@ def get_trainer_params(
plugins = []
if accelerator:
plugins.append(
DDPPlugin(find_unused_parameters=cfg.MODEL.DDP_FIND_UNUSED_PARAMETERS)
DDPStrategy(find_unused_parameters=cfg.MODEL.DDP_FIND_UNUSED_PARAMETERS)
)
return {
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment