Commit 46f16a5e authored by Yuxin Wu's avatar Yuxin Wu Committed by Facebook GitHub Bot
Browse files

update benchmark_storage with instructions

Summary: Also modify launch() because it should not assume it's always called with a CfgNode object.

Differential Revision: D31494215

fbshipit-source-id: 8f07e9cb64969f8a14641956f7ef7c7160748bd9
parent 8bba1fc6
...@@ -64,8 +64,7 @@ def launch( ...@@ -64,8 +64,7 @@ def launch(
if get_launch_environment() == "local" and not torch.cuda.is_available(): if get_launch_environment() == "local" and not torch.cuda.is_available():
assert len(args) > 0, args assert len(args) > 0, args
cfg = args[0] cfg = args[0]
assert isinstance(cfg, CfgNode) if isinstance(cfg, CfgNode) and cfg.MODEL.DEVICE == "cuda":
if cfg.MODEL.DEVICE == "cuda":
logger.warning( logger.warning(
"Detected that CUDA is not available on this machine, set MODEL.DEVICE" "Detected that CUDA is not available on this machine, set MODEL.DEVICE"
" to cpu and backend to GLOO" " to cpu and backend to GLOO"
......
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