Upgrade torchvision

Stopgap fix for #551 Signed-off-by: Ben Firshman <[email protected]>
bfirsh · Mar 11, 2021 · 86fb7ab · 86fb7ab
1 parent fe817e2
commit 86fb7ab
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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