diff --git a/requirements.txt b/requirements.txt
index 173dffe..17cf171 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -16,7 +16,7 @@ MarkupSafe==1.1.0
 numpy==1.16.1
 Pillow==5.4.1
 protobuf==3.6.1
-PyYAML==3.13
+PyYAML==5.4
 scipy==1.2.1
 six==1.12.0
 tensorboard==1.12.2