diff --git a/requirements.txt b/requirements.txt
index 6c05ce1..fc2fc37 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -2,7 +2,7 @@ torch==1.4.0
 numpy==1.18.1
 matplotlib==3.1.2
 wandb==0.8.21
-pyyaml==5.3
+pyyaml==5.4
 Pillow==7.0.0
 scikit-learn==0.22.1
 h5py==2.10.0