- 21 May, 2019 1 commit
-
-
Francisco Massa authored
-
- 19 May, 2019 2 commits
-
-
ekka authored
* Remove dependency from functool in ShuffleNetsV2 This PR removes the dependence of the ShuffleNetV2 code from `functool`. * flake fix
-
Francisco Massa authored
Also move weights from ShuffleNet to PyTorch bucket. Additionally, rename shufflenet to make it consistent with the other models
-
- 08 May, 2019 1 commit
-
-
Bar authored
* Enhance ShufflenetV2 Class shufflenetv2 receives `stages_repeats` and `stages_out_channels` arguments. * remove explicit num_classes argument from utility functions
-
- 07 May, 2019 1 commit
-
-
ekka authored
* Minor refactoring of ShuffleNetV2 Added progress flag following #875. Further the following refactoring was also done: 1) added `version` argument in shufflenetv2 method and removed the operations for converting the `width_mult` arg to float and string. 2) removed `num_classes` argument and **kwargs from functions except `ShuffleNetV2` * removed `version` arg * Update shufflenetv2.py * Removed the try except block * Update shufflenetv2.py * Changed version from float to str * Replace `width_mult` with `stages_out_channels` Removes the need of `_getStages` function.
-
- 06 May, 2019 1 commit
-
-
ekka authored
* remove 'input_size' parameter from shufflenetv2 * Update shufflenetv2.py
-
- 30 Apr, 2019 1 commit
-
-
Bar authored
* Add ShuffleNet v2 Added 4 configurations: x0.5, x1, x1.5, x2 Add 2 pretrained models: x0.5, x1 * fix lint * Change globalpool to torch.mean() call
-