diff --git a/requirements.txt b/requirements.txt
index f0cc1c6b72ba60dcb9b27bfc2b2245929a27a881..3d31a4ac4ef41c1aefb11b10a2adee459d3fce4e 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -2,3 +2,4 @@ paddlepaddle >= 1.2.0
 pre-commit
 protobuf >= 3.1.0
 yapf == 0.26.0
+pyyaml