diff --git a/requirements.txt b/requirements.txt
index 8e81bc6..ee93d76 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -3,7 +3,7 @@ torchsummary==1.5.1
 torchvision==0.3.0
 numpy==1.17.0
 Pillow==6.0.0
-PyYAML==5.1.2
+PyYAML==5.4
 requests==2.22.0
 tqdm==4.38.0
 opencv-python