- 01 Oct, 2021 1 commit
-
-
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>
-
- 13 Sep, 2021 1 commit
-
-
Philip Meier authored
* add pre-commit hooks * ignore yamls in packaging/* * add pre-commit to contributing guide lines * Update CONTRIBUTING.md Co-authored-by:
Nicolas Hug <contact@nicolas-hug.com> * remove some hooks * fix docstrings * fix end of files Co-authored-by:
Nicolas Hug <contact@nicolas-hug.com>
-
- 07 Sep, 2021 1 commit
-
-
Alexander Soare authored
-
- 06 Sep, 2021 1 commit
-
-
Alexander Soare authored
* add fx feature extraction util * Make it possible to use train and eval mode * FX feature extraction - Tweaks and small bug fixes * FX feature extraction - add tests * move to feature_extraction.py, add LeafModuleAwareTracer, add docs * Tweaks to docs * addressing latest round of feedback * undo line spacing changes * change type hints in docstrings * fix sphinx indentation * expose feature_extraction * add maskrcnn example * add api refernce subheading * address latest review notes, refactor names, fix regex, cosmetics * Add back efficientnet to models * fix tests for effnet * fix linting issue * fix test tracer kwargs Co-authored-by:Francisco Massa <fvsmassa@gmail.com>
-