Fix error when running on cpu with MODEL.DEVICE="cpu"

Summary: The assert just below fails because `backend = "NCCL"` and we don't have a GPU Reviewed By: ppwwyyxx Differential Revision: D31506095 fbshipit-source-id: c1199eeb732d098c02fe5cd40efb85284deaa3b9

Fix error when running on cpu with MODEL.DEVICE="cpu"
Summary: The assert just below fails because `backend = "NCCL"` and we don't have a GPU Reviewed By: ppwwyyxx Differential Revision: D31506095 fbshipit-source-id: c1199eeb732d098c02fe5cd40efb85284deaa3b9
8bba1fc6 · Daniel Haziza · Facebook GitHub Bot · 004be54f · 8bba1fc6
Commit 8bba1fc6 authored Oct 13, 2021 by Daniel Haziza Committed by Facebook GitHub Bot Oct 13, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

d2go/distributed.py d2go/distributed.py +1 -1

No files found.
--- a/d2go/distributed.py
+++ b/d2go/distributed.py
@@ -72,7 +72,7 @@ def launch(
            )
            with temp_defrost(cfg):
                cfg.MODEL.DEVICE = "cpu"
-            backend = "GLOO"
+        backend = "GLOO"

    if backend == "NCCL":
        assert (