--- a
+++ b/main.py
@@ -0,0 +1,15 @@
+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()