• 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
presets.py 1.52 KB