Merge pull request #268 from seemethere/patch-1

Update torch -> 1.7.0, torchvision -> 0.8.1 for github actions
4 years ago · 2542283b28
parent 53aeed3499 ab9f1fc995
commit 2542283b28
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: