diff --git a/setup.py b/setup.py index ead6d77..7714631 100644 --- a/setup.py +++ b/setup.py @@ -17,7 +17,7 @@ "attrs", "multiprocess", "opencv-python", - "tensorflow==2.3.1", + "tensorflow==2.11.1", "torch", ], scripts=['cluster/holy_dannce_train.sh',