1. 02 Jul, 2019 1 commit
    • 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
  2. 07 Jun, 2019 1 commit
  3. 19 May, 2019 1 commit
  4. 02 Apr, 2019 1 commit
  5. 28 Mar, 2019 1 commit
    • Francisco Massa's avatar
      Add MobileNet V2 (#818) · a61803f0
      Francisco Massa authored
      * Add MobileNet V2
      
      * Remove redundant functions and make tests pass
      
      * Simplify a bit the implementation
      
      * Reuse ConvBNReLU more often
      
      * Remove input_size and minor changes
      
      * Py2 fix
      a61803f0