diff --git a/requirements.txt b/requirements.txt
index ab7c421..74784f6 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -4,4 +4,4 @@ torch==0.4.0
 wget==3.2
 Pillow==5.1.0
 tensorboardX==1.2
-PyYAML==4.2b1
+PyYAML==5.4