diff --git a/requirements.txt b/requirements.txt
index 41806ca..be36d98 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -11,7 +11,7 @@ protobuf
 PyYAML
 scipy
 six
-tensorflow==2.1
+tensorflow==2.4.0
 termcolor
 tqdm
 Werkzeug