diff --git a/requirements-test.txt b/requirements-test.txt index df134308d..230cc84d4 100644 --- a/requirements-test.txt +++ b/requirements-test.txt @@ -3,8 +3,8 @@ black==19.10b0 pytest==6.1.1 tensorflow-cpu==2.4.0 pytorch-lightning==1.1.1 -torch==1.4.0 -torchvision==0.5.0 +torch==1.8.0 +torchvision==0.9.0 # for integration tests boto3==1.12.32 google-cloud-storage==1.32.0