diff --git a/requirements.txt b/requirements.txt
index 78b760c..5d38f93 100755
--- a/requirements.txt
+++ b/requirements.txt
@@ -6,7 +6,7 @@ torch >= 1.5
 matplotlib
 pillow
 tensorboard
-pyyaml
+pyyaml >= 5.3
 torchvision
 scipy
 tqdm