1. 01 Oct, 2021 1 commit
  2. 13 Sep, 2021 1 commit
  3. 07 Sep, 2021 1 commit
  4. 06 Sep, 2021 1 commit
    • Alexander Soare's avatar
      Add FX feature extraction as an alternative to intermediate_layer_getter (#4302) · 72d650ae
      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: default avatarFrancisco Massa <fvsmassa@gmail.com>
      72d650ae