Merge pull request #125 from FrancescoSaverioZuppichini/develop

Develop

.github/workflows/doc_builder.yml

@@ -24,7 +24,8 @@ jobs:
     - name: Install dependencies
       run: |
         python -m pip install --upgrade pip
-        pip install sphinx sphinx_rtd_theme recommonmark torch torchvision
+        pip install sphinx sphinx_rtd_theme recommonmark torch torchvision pandas tqdm torchsummary timm pretrainedmodels
+
     - name: Build doc
       run: |
         cd ./docs

.github/workflows/test_runner.yml

@@ -25,7 +25,8 @@ jobs:
     - name: Install dependencies
       run: |
         python -m pip install --upgrade pip
-        pip install flake8 pytest  torch torchvision coverage requests tqdm
+        pip install flake8 pytest  torch torchvision coverage requests tqdm timm pretrainedmodels efficientnet_pytorch torchsummary
+        pip install git+https://github.com/sksq96/pytorch-summary.git
         if [ -f requirements.txt ]; then pip install -r requirements.txt; fi
     - name: Lint with flake8
       run: |

.gitignore

@@ -1,6 +1,5 @@
 docs/_build/
 #
-*.pth
 .vscode/
 # Byte-compiled / optimized / DLL files
 __pycache__/
@@ -131,3 +130,5 @@ dmypy.json
 
 # Pyre type checker
 .pyre/
+
+table-torchvision.md