diff --git a/requirements.txt b/requirements.txt
index b545c3e..b7ca494 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -4,5 +4,5 @@ opencv-python==3.4.1.15
 scipy==1.1.0
 pillow==5.1.0
 tensorboardX
-pyyaml==3.12
+pyyaml==5.1
 tqdm==4.28.1