diff --git a/requirements.txt b/requirements.txt index 190c955..7506c9c 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,3 +1,3 @@ -tensorflow==1.13.1 +tensorflow==2.7.2 horovod==0.18.1 gradient \ No newline at end of file