config_generator.py

import yaml
import utils
import data


def generate_config(
        template_config,
        env_nums,
        n_rotations_list,
        movement_modes,
        model_names,
    ):
    template = utils.load_config(template_config)
    for env_num in env_nums:
        for n_rotations in n_rotations_list:
            for movement_mode in movement_modes:
                for model_name in model_names:
                    output_layers = data.load_model_layers(model_name)
                    for output_layer in output_layers:
                        config_version = \
                            f'env{env_num}_r{n_rotations}_{movement_mode}_' \
                            f'{model_name}_{output_layer}'
                        
                        template['config_version'] = config_version
                        template['unity_env'] = f'env{env_num}_r{n_rotations}'
                        template['n_rotations'] = n_rotations
                        template['movement_mode'] = movement_mode
                        template['model_name'] = model_name
                        template['output_layer'] = output_layer
                        with open(f'configs/{config_version}.yaml', 'w') as f:
                            yaml.dump(template, f, sort_keys=False)


if __name__ == "__main__":
    generate_config(
        template_config='env28_r24_2d_vgg16_fc2',
        env_nums=['28run2', '37', '39', '40'],
        n_rotations_list=[24],
        movement_modes=['2d'],
        model_names=['vgg16'],
    )