Commit 8bba1fc6 authored by Daniel Haziza's avatar Daniel Haziza Committed by Facebook GitHub Bot
Browse files

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
parent 004be54f
......@@ -72,7 +72,7 @@ def launch(
)
with temp_defrost(cfg):
cfg.MODEL.DEVICE = "cpu"
backend = "GLOO"
backend = "GLOO"
if backend == "NCCL":
assert (
......
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