- 02 Jul, 2019 1 commit
-
-
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
-
- 07 Jun, 2019 1 commit
-
-
Matthew Yeung authored
* allow user to define residual settings * 4spaces * linting errors * backward compatible, and added test
-
- 19 May, 2019 1 commit
-
-
Francisco Massa authored
Also move weights from ShuffleNet to PyTorch bucket. Additionally, rename shufflenet to make it consistent with the other models
-
- 02 Apr, 2019 1 commit
-
-
Surgan Jandial authored
Making references/classification/train.py and references/classification/utils.py compatible with python2 (#831) * linter fixes * linter fixes
-
- 28 Mar, 2019 1 commit
-
-
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
-