Move config out of module

config/dgx.py

@@ -1,5 +1,9 @@
 import ml_collections
-from ddpo_pytorch.config import base
+import imp
+import os
+
+base = imp.load_source("base", os.path.join(os.path.dirname(__file__), "base.py"))
+
 
 def get_config():
     config = base.get_config()
@@ -20,4 +24,4 @@ def get_config():
 
     config.per_prompt_stat_tracking = None
 
-    return config
+    return config

scripts/train.py

@@ -26,7 +26,7 @@ tqdm = partial(tqdm.tqdm, dynamic_ncols=True)
 
 
 FLAGS = flags.FLAGS
-config_flags.DEFINE_config_file("config", "ddpo_pytorch/config/base.py", "Training configuration.")
+config_flags.DEFINE_config_file("config", "config/base.py", "Training configuration.")
 
 logger = get_logger(__name__)