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 .

Citation

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

@article{SSS2017
  author =   {Zehao Huang and Naiyan Wang},
  title =    {Data-Driven Sparse Structure Selection for Deep Neural Networks},
  journal =  {arXiv preprint arXiv:1707.01213},
  year =     {2017}
}

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