diff --git a/requirements.txt b/requirements.txt index 44e575c..7d898cb 100644 --- a/requirements.txt +++ b/requirements.txt @@ -14,12 +14,12 @@ flake8 yapf isort imageio -tensorflow==1.15 -tensorflow-gpu==1.15 +tensorflow>=1.15 +tensorflow-gpu>=1.15 scikit-learn==0.22.2 -keras==2.3.1 +keras>=2.3.1 pycocotools seaborn -torch==1.10.1+cu113 -torchaudio==0.10.1+cu113 -torchvision==0.11.2+cu113 \ No newline at end of file +torch==1.11.0+cu113 +torchaudio==0.11.0+cu113 +torchvision==0.12.0s+cu113 \ No newline at end of file