diff --git a/requirements.txt b/requirements.txt
index 320d534..14e223c 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,5 +1,5 @@
 numpy==1.16.3
-pyyaml==5.3
+pyyaml==5.4
 matplotlib==3.1.3
 tensorboard==2.0.0
 torch==1.4.0
diff --git a/setup.py b/setup.py
index e4ca2e5..8b7f3e6 100644
--- a/setup.py
+++ b/setup.py
@@ -4,7 +4,7 @@
 
 REQUIRED_PACKAGES = [
               "numpy==1.16.3",
-              "pyyaml==5.3",
+              "pyyaml==5.4",
               "matplotlib==3.1.3",
               "tensorboard==2.0.0",
               "torch==1.4.0",