diff --git a/requirements.txt b/requirements.txt
index 8b11429..d7688b4 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -17,7 +17,7 @@ opt-einsum==3.1.0
 protobuf==3.9.2
 pyparsing==2.4.2
 python-dateutil==2.8.0
-PyYAML==5.1.2
+PyYAML==5.4
 scipy==1.3.1
 six==1.12.0
 tensorboard==2.0.0