1. 21 May, 2019 1 commit
  2. 19 May, 2019 2 commits
  3. 08 May, 2019 1 commit
    • Bar's avatar
      Enhance ShufflenetV2 (#892) · 43ab2fef
      Bar authored
      * Enhance ShufflenetV2
      
      Class shufflenetv2 receives `stages_repeats` and `stages_out_channels` arguments.
      
      * remove explicit num_classes argument from utility functions
      43ab2fef
  4. 07 May, 2019 1 commit
    • ekka's avatar
      Refactoring of ShuffleNetV2 (#889) · 0564df43
      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.
      0564df43
  5. 06 May, 2019 1 commit
  6. 30 Apr, 2019 1 commit
    • Bar's avatar
      Add ShuffleNet v2 (#849) · 7a4845a9
      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
      7a4845a9