def get_container_name(framework):
    """Translate the framework into the docker container name."""
    return 'baseline-{}'.format({
        'tensorflow': 'tf',
        'pytorch': 'pyt',
        'dynet': 'dy'
    }[framework])