File size: 1,313 Bytes
7cdf421 |
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 |
import yaml
def load_model_config(stage, mode):
# load special config for each model
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 config
base_configuration = load_base_config()
# load stage config
# if args.get('mode'):
stage_configuration = load_model_config(args['stage'], args['mode'])
# update and append the stage config for base config
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
|