ShuffleNet model - CeLuigi/models-comparison.pytorch GitHub Wiki

X. Zhang, X. Zhou, M. Lin, and J. Sun, “Shufflenet: An extremely efficient convolutional neural network for mobile devices,” in Conference on Computer Vision and Pattern Recognition (CVPR). IEEE, 2018.

@InProceedings{Zhang_2018_CVPR,
author = {Zhang, Xiangyu and Zhou, Xinyu and Lin, Mengxiao and Sun, Jian},
title = {ShuffleNet: An Extremely Efficient Convolutional Neural Network for Mobile Devices},
booktitle = {Conference on Computer Vision and Pattern Recognition (CVPR)},
year = {2018},
organization={IEEE}
}

Source: ShuffleNet repo

• shufflenet(num_classes=1000, pretrained='imagenet')