diff --git a/setup.py b/setup.py index 7a62e2e..94ba0e8 100644 --- a/setup.py +++ b/setup.py @@ -30,7 +30,7 @@ 'scikit-learn>=0.19.1', 'scikit-neuralnetwork>=0.7', 'scipy==1.0.0', - 'tensorflow==1.13.1', + 'tensorflow==2.5.3', 'opencv-python', 'h5py', 'pydot',