| import yaml |
|
|
|
|
| def load_model_config(stage, mode): |
| |
| config_path = f'config/stage_{stage}.yaml' |
| print(f'[!] load configuration from {config_path}') |
| with open(config_path) as f: |
| configuration = yaml.load(f, Loader=yaml.FullLoader) |
| new_config = {} |
| for key, value in configuration.items(): |
| if key in ['train', 'test', 'validation']: |
| if mode == key: |
| new_config.update(value) |
| else: |
| new_config[key] = value |
| configuration = new_config |
| return configuration |
|
|
|
|
| def load_config(args): |
| '''the configuration of each model can rewrite the base configuration''' |
| |
| base_configuration = load_base_config() |
|
|
| |
| |
| stage_configuration = load_model_config(args['stage'], args['mode']) |
|
|
| |
| base_configuration.update(stage_configuration) |
| configuration = base_configuration |
| return configuration |
|
|
|
|
| def load_base_config(): |
| config_path = f'config/base.yaml' |
| with open(config_path) as f: |
| configuration = yaml.load(f, Loader=yaml.FullLoader) |
| print(f'[!] load base configuration: {config_path}') |
| return configuration |
|
|
