diff --git a/requirements.txt b/requirements.txt
index c2cb824..49f1614 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,4 +1,4 @@
 progressbar2==3.34.2
-PyYAML==3.13
+PyYAML==5.4
 numpy==1.13.1
 tensorflow==1.3.0