Running with D2Go

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

Reviewed By: tglik

Differential Revision: D41714933

fbshipit-source-id: 5b2b3610af554f6082a4025af0673b4bc34b17ca

d2go/config/__init__.py

@@ -4,6 +4,7 @@
 
 # forward the namespace to avoid `d2go.config.config`
 from .config import (
+    add_cfg_nodes,
     auto_scale_world_size,
     CfgNode,
     CONFIG_CUSTOM_PARSE_REGISTRY,
@@ -18,6 +19,7 @@ __all__ = [
     "CONFIG_CUSTOM_PARSE_REGISTRY",
     "CONFIG_SCALING_METHOD_REGISTRY",
     "CfgNode",
+    "add_cfg_nodes",
     "auto_scale_world_size",
     "load_full_config_from_file",
     "reroute_config_path",

d2go/config/config.py

@@ -4,7 +4,7 @@
 import contextlib
 import copy
 import logging
-from typing import List
+from typing import Dict, List
 
 import mock
 import yaml
@@ -183,3 +183,13 @@ def load_full_config_from_file(filename: str) -> CfgNode:
     cfg = loaded_cfg.get_default_cfg()
     cfg.merge_from_other_cfg(loaded_cfg)
     return cfg
+
+
+def add_cfg_nodes(cfg, key: str, cfg_dict: Dict):
+    node = CfgNode()
+    setattr(cfg, key, node)
+    for k, v in cfg_dict.items():
+        if isinstance(v, dict):
+            add_cfg_nodes(node, k, v)
+        else:
+            setattr(node, k, v)