| import yaml | |
| # https://github.com/pfnet-research/sngan_projection/blob/master/source/yaml_utils.py | |
| class Config(object): | |
| def __init__(self, config_dict): | |
| self.config = config_dict | |
| def __getattr__(self, key): | |
| if key in self.config: | |
| return self.config[key] | |
| else: | |
| raise AttributeError(key) | |
| def __getitem__(self, key): | |
| return self.config[key] | |
| def __repr__(self): | |
| return yaml.dump(self.config, default_flow_style=False) | |