Skip to content

GitLab

"tests/scripts/vscode:/vscode.git/clone" did not exist on "e4e0132972a09df8f73387c4ea48d2cd19178d2a"
Commit 00409af8 authored by Sam Tsai's avatar Sam Tsai Committed by Facebook GitHub Bot
Browse files

lightning - deprecating distributed backend, switching to use_ddp 

Summary:
Pull Request resolved: https://github.com/facebookresearch/d2go/pull/180

Distributed backend is deprecated. Switching to use "use_ddp" instead.

Reviewed By: kazhang

Differential Revision: D34394993

fbshipit-source-id: a5bfb22f8952d20c9a8d86322cd740534c25c689
parent eee4dfc1
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 4 deletions
d2go/runner/lightning_task.py
View file @ 00409af8
...@@ -259,10 +259,9 @@ class DefaultTask(pl.LightningModule): ...@@ -259,10 +259,9 @@ 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 len(self.trainer._accelerator_connector.parallel_devices) == 1
or self.trainer._accelerator_connector.distributed_backend.lower() or self.trainer._accelerator_connector.use_ddp
in ["ddp", "ddp_cpu"] ), "Only DDP and DDP_CPU distributed backend are supported"
), ("Only DDP and DDP_CPU distributed backend are supported")
def _get_inference_dir_name( def _get_inference_dir_name(
base_dir, inference_type, dataset_name, model_tag: ModelTag base_dir, inference_type, dataset_name, model_tag: ModelTag
......
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