diff --git a/requirements.txt b/requirements.txt
index 2958643b..688dbd7b 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,5 +1,5 @@
 # Main training requirements
-tensorflow == 1.13.1
+tensorflow == 2.6.4
 numpy == 1.15.4
 progressbar2
 pandas