Update torch -> 1.7.0, torchvision -> 0.8.1

4 years ago · ab9f1fc995
parent 53aeed3499
commit ab9f1fc995
1 changed files with 2 additions and 2 deletions
--- a/.github/workflows/tests.yml
+++ b/.github/workflows/tests.yml
@ -17,8 +17,8 @@ jobs:
      matrix:
        os: [ubuntu-latest, macOS-latest]
        python: ['3.8']
-        torch: ['1.5.0']
-        torchvision: ['0.6.0']
+        torch: ['1.7.0']
+        torchvision: ['0.8.1']
    runs-on: ${{ matrix.os }}

    steps: