diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml index c300787..6c4fcca 100644 --- a/.github/workflows/tests.yml +++ b/.github/workflows/tests.yml @@ -13,7 +13,7 @@ jobs: fail-fast: false matrix: python-version: ["3.8", "3.9"] - torch-version: ["1.8", "1.12"] + torch-version: ["1.12", "2.0"] steps: - uses: actions/checkout@v3 diff --git a/requirements.txt b/requirements.txt index 79be438..f5bb4fe 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,2 +1,2 @@ numpy>=1.20.0 -torch>=1.8.0 +torch>=1.12.0