add damit uri support in train_net local run

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

as title

Reviewed By: jiaxuzhu92, ywwwer

Differential Revision: D49672980

fbshipit-source-id: f34ffe944c25c948fe1abd492ea0b96e47dc5b06

d2go/runner/config_defaults.py

@@ -219,3 +219,8 @@ def get_generalized_rcnn_runner_default_cfg(cfg: CN) -> CN:
     _add_rcnn_default_config(cfg)
 
     return cfg
+
+
+@fb_overwritable()
+def preprocess_cfg(cfg: CN) -> CN:
+    return cfg

tools/train_net.py

@@ -13,6 +13,7 @@ import detectron2.utils.comm as comm
 from d2go.config import CfgNode
 from d2go.distributed import distributed_worker, launch
 from d2go.runner import BaseRunner
+from d2go.runner.config_defaults import preprocess_cfg
 from d2go.setup import (
     basic_argument_parser,
     build_basic_cli_args,
@@ -112,6 +113,7 @@ def wrapped_main(*args, **kwargs) -> Callable[..., TrainOrTestNetOutput]:
 
 def run_with_cmdline_args(args):
     cfg, output_dir, runner_name = prepare_for_launch(args)
+    cfg = preprocess_cfg(cfg)
     shared_context = setup_before_launch(cfg, output_dir, runner_name)
 
     main_func = (