Commit 256b3f47 authored by Miquel Jubert Hermoso's avatar Miquel Jubert Hermoso Committed by Facebook GitHub Bot
Browse files

Add option to specify environment variables for MAST in the extra args field

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

Reviewed By: frabu6

Differential Revision: D43538416

fbshipit-source-id: ca323dff0ae3b0dee1eccac3c4f8b95d01c6e5d9
parent 4dc7e94c
...@@ -33,6 +33,11 @@ def _add_detectron2go_runner_default_fb_cfg(_C: CN) -> None: ...@@ -33,6 +33,11 @@ def _add_detectron2go_runner_default_fb_cfg(_C: CN) -> None:
pass pass
@fb_overwritable()
def _add_base_runner_default_fb_cfg(_C: CN) -> None:
pass
def _add_detectron2go_runner_default_cfg(_C: CN) -> None: def _add_detectron2go_runner_default_cfg(_C: CN) -> None:
# _C.MODEL.FBNET_V2... # _C.MODEL.FBNET_V2...
add_fbnet_v2_default_configs(_C) add_fbnet_v2_default_configs(_C)
...@@ -146,6 +151,8 @@ def get_base_runner_default_cfg(cfg: CN) -> CN: ...@@ -146,6 +151,8 @@ def get_base_runner_default_cfg(cfg: CN) -> CN:
# run evaluation after training is done # run evaluation after training is done
cfg.TEST.FINAL_EVAL = True cfg.TEST.FINAL_EVAL = True
_add_base_runner_default_fb_cfg(cfg)
return cfg return cfg
......
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