diff --git a/requirements.txt b/requirements.txt
index 588bf64..3a79006 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -11,7 +11,7 @@ protobuf
 PyYAML
 scipy
 six
-tensorflow==2.1
+tensorflow==2.44
 termcolor
 tqdm
 Werkzeug