diff --git a/requirements.txt b/requirements.txt
index 83569f0..91e1b23 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -6,7 +6,7 @@ idna==2.7
 numpy==1.15.0
 pkg-resources==0.0.0
 pyglet==1.3.2
-PyYAML==5.1
+PyYAML==5.4
 six==1.11.0
 torch==0.3.1
 urllib3==1.24.2