iicd/ddpo-pytorch
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Move config out of module

Browse Source
This commit is contained in:
Kevin Black 2023-06-25 21:02:27 -07:00
parent 269615a35e
commit 5c16a90ceb
3 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

0
ddpo_pytorch/config/base.py → config/base.py
View File

6
ddpo_pytorch/config/dgx.py → config/dgx.py
View File

@ -1,5 +1,9 @@
import ml_collections 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(): def get_config():
config = base.get_config() config = base.get_config()

2
scripts/train.py
View File

@ -26,7 +26,7 @@ tqdm = partial(tqdm.tqdm, dynamic_ncols=True)
FLAGS = flags.FLAGS 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__) logger = get_logger(__name__)