diff --git a/setup.py b/setup.py index 28f986a..fd6551d 100644 --- a/setup.py +++ b/setup.py @@ -1,8 +1,8 @@ from setuptools import find_packages, setup dependencies = [ - 'tensorflow==2.2.1', - 'numpy==1.18.4', + 'tensorflow==2.5.0', + 'numpy>=1.18.4', 'tensorflow-datasets==3.1.0' ]