• 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
shufflenetv2.py 5.49 KB