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

Mark accelerator connector as protected (#10032)

Summary:
### New commit log messages
  1f7bd6650 Mark accelerator connector as protected (#10032)

Reviewed By: yifuwang

Differential Revision: D31905981

fbshipit-source-id: a7f0f03033b02b603d28203ae2c8e8df4933fb23
parent d528bc11
...@@ -237,8 +237,8 @@ class DefaultTask(pl.LightningModule): ...@@ -237,8 +237,8 @@ class DefaultTask(pl.LightningModule):
def _reset_dataset_evaluators(self): def _reset_dataset_evaluators(self):
"""reset validation dataset evaluator to be run in EVAL_PERIOD steps""" """reset validation dataset evaluator to be run in EVAL_PERIOD steps"""
assert ( assert (
not self.trainer.accelerator_connector.distributed_backend not self.trainer._accelerator_connector.distributed_backend
or self.trainer.accelerator_connector.distributed_backend.lower() or self.trainer._accelerator_connector.distributed_backend.lower()
in ["ddp", "ddp_cpu"] in ["ddp", "ddp_cpu"]
), ("Only DDP and DDP_CPU distributed backend are supported") ), ("Only DDP and DDP_CPU distributed backend are supported")
......
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