Skip to content
No description, website, or topics provided.
Python
Branch: master
Clone or download
Latest commit 12ed0fd Jul 17, 2018
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
config update Apr 19, 2018
core init Apr 19, 2018
data init Apr 19, 2018
docs update fig Apr 19, 2018
incubator-mxnet-bk @ 59ba595 add submodule Apr 19, 2018
symbol init Apr 19, 2018
.gitignore Initial commit Apr 19, 2018
.gitmodules add submodule Apr 19, 2018
LICENSE Initial commit Apr 19, 2018
README.md Update README.md Jul 17, 2018
train.py init Apr 19, 2018

README.md

sparse-structure-selection

This code is a re-implementation of the imagenet classification experiments in the paper Data-Driven Sparse Structure Selection for Deep Neural Networks (ECCV2018).

Citation

If you use our code in your research or wish to refer to the baseline results, please use the following BibTeX entry.

@article{SSS2018
  author =   {Zehao Huang and Naiyan Wang},
  title =    {Data-Driven Sparse Structure Selection for Deep Neural Networks},
  journal =  {ECCV},
  year =     {2018}
}

Implementation

This code is implemented by a modified MXNet which supports ResNeXt-like augmentation. (This version of MXNet does not support cudnn7)

ImageNet data preparation

Download the ImageNet dataset and create pass through rec (following tornadomeet's repository but using unchange mode)

Run

  • modify config/cfgs.py
  • python train.py

Results on ImageNet-1k

You can’t perform that action at this time.