from src import get_config, train, evaluate, test

# Get config from conf.yaml

conf = get_config('./conf/evaluation.yaml')

task = conf['task']

if task == 'training':

    train(conf)

elif task == 'evaluation':

    evaluate(conf)

elif task == 'testing':

    test(conf)

else:

    print('Task not supported.')

    exit()