• yaysummeriscoming's avatar
    Fixed width multiplier (#1005) · 8350645b
    yaysummeriscoming authored
    * Fixed width multiplier
    
    Layer channels are now rounded to a multiple of 8, as per the official tensorflow implementation.  I found this fix when looking through: https://github.com/d-li14/mobilenetv2.pytorch
    
    * Channel multiple now a user configurable option
    
    The official tensorflow slim mobilenet v2 implementation rounds the number of channels in each layer to a multiple of 8.  This is now user configurable - 1 turns off rounding
    
    * Fixed whitespace error
    
    Fixed error: ./torchvision/models/mobilenet.py:152:1: W293 blank line contains whitespace
    8350645b
mobilenet.py 5.73 KB