Spaces:
Sleeping
Sleeping
File size: 1,492 Bytes
03022ee | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 | import os
import yaml
class Config(object):
def __init__(self, conf_dict):
for key, value in conf_dict.items():
self.__dict__[key] = value
def convert_to_yaml(overrides):
"""Convert args to yaml for overrides"""
yaml_string = ""
# Handle '--arg=val' type args
joined_args = "=".join(overrides)
split_args = joined_args.split("=")
for arg in split_args:
if arg.startswith("--"):
yaml_string += "\n" + arg[len("--") :] + ":"
else:
yaml_string += " " + arg
return yaml_string.strip()
def yaml_config_loader(conf_file, overrides=None):
with open(conf_file, "r") as fr:
conf_dict = yaml.load(fr, Loader=yaml.FullLoader)
if overrides is not None:
overrides = yaml.load(overrides, Loader=yaml.FullLoader)
conf_dict.update(overrides)
return conf_dict
def build_config(config_file, overrides=None, copy=False):
if config_file.endswith(".yaml"):
if overrides is not None:
overrides = convert_to_yaml(overrides)
conf_dict = yaml_config_loader(config_file, overrides)
if copy and 'exp_dir' in conf_dict:
os.makedirs(conf_dict['exp_dir'], exist_ok=True)
saved_path = os.path.join(conf_dict['exp_dir'], 'config.yaml')
with open(saved_path, 'w') as f:
f.write(yaml.dump(conf_dict))
else:
raise ValueError("Unknown config file format")
return Config(conf_dict)
|