-
Alexander Soare authored
* draft commit * Polish and add corresponding test * Update docs * Update torchvision/models/feature_extraction.py * Update docs/source/feature_extraction.rst Co-authored-by:Francisco Massa <fvsmassa@gmail.com>
903ea4a6
* draft commit
* Polish and add corresponding test
* Update docs
* Update torchvision/models/feature_extraction.py
* Update docs/source/feature_extraction.rst
Co-authored-by: Francisco Massa <fvsmassa@gmail.com>