Test all supported versions of PyTorch (#124)

* Test all supported versions of PyTorch

Author: TylerYep

.github/workflows/test.yml

@@ -12,6 +12,14 @@ jobs:
       fail-fast: false
       matrix:
         python-version: ["3.7", "3.8", "3.9"]
+        pytorch-version: ["1.4.0", "1.5.1", "1.6.0", "1.7.1", "1.8", "1.9", "1.10", "1.11"]
+        exclude:
+          - python-version: 3.9
+            pytorch-version: 1.6.0
+          - python-version: 3.9
+            pytorch-version: 1.5.1
+          - python-version: 3.9
+            pytorch-version: 1.4.0
 
     steps:
     - uses: actions/checkout@v2
@@ -23,13 +31,18 @@ jobs:
       run: |
         python -m pip install --upgrade pip
         python -m pip install mypy pytest pytest-cov
-        pip install -r requirements.txt
+        pip install torch==${{ matrix.pytorch-version }} torchvision
     - name: mypy
+      if: ${{ matrix.pytorch-version == '1.11' }}
       run: |
         mypy .
     - name: pytest
-      if: always()
+      if: ${{ matrix.pytorch-version == '1.11' }}
       run: |
         pytest --cov=torchinfo --cov-report= --durations=0
+    - name: pytest
+      if: ${{ matrix.pytorch-version != '1.11' }}
+      run: |
+        pytest --no-output -k "not test_eval_order_doesnt_matter and not test_google and not test_uninitialized_tensor and not test_input_size_half_precision"
     - name: codecov
       uses: codecov/codecov-action@v1

README.md

@@ -15,6 +15,8 @@ Torchinfo provides information complementary to what is provided by `print(your_
 
 This is a completely rewritten version of the original torchsummary and torchsummaryX projects by @sksq96 and @nmhkahn. This project addresses all of the issues and pull requests left on the original projects by introducing a completely new API.
 
+Supports PyTorch versions 1.4.0+.
+
 # Usage
 
 ```