1. 26 Aug, 2021 1 commit
    • Vasilis Vryniotis's avatar
      Add EfficientNet Architecture in TorchVision (#4293) · 37a9ee5b
      Vasilis Vryniotis authored
      * Adding code skeleton
      
      * Adding MBConvConfig.
      
      * Extend SqueezeExcitation to support custom min_value and activation.
      
      * Implement MBConv.
      
      * Replace stochastic_depth with operator.
      
      * Adding the rest of the EfficientNet implementation
      
      * Update torchvision/models/efficientnet.py
      
      * Replacing 1st activation of SE with SiLU.
      
      * Adding efficientnet_b3.
      
      * Replace mobilenetv3 assets with custom.
      
      * Switch to standard sigmoid and reconfiguring BN.
      
      * Reconfiguration of efficientnet.
      
      * Add repr
      
      * Add weights.
      
      * Update weights.
      
      * Adding B5-B7 weights.
      
      * Update docs and hubconf.
      
      * Fix doc link.
      
      * Fix typo on comment.
      37a9ee5b
  2. 28 Jan, 2021 1 commit
    • Vasilis Vryniotis's avatar
      Adding Preset Transforms in reference scripts (#3317) · 1703e4ca
      Vasilis Vryniotis authored
      * Adding presets in the classification reference scripts.
      
      * Adding presets in the object detection reference scripts.
      
      * Adding presets in the segmentation reference scripts.
      
      * Adding presets in the video classification reference scripts.
      
      * Moving flip at the end to align with image classification signature.
      1703e4ca