ShuffleNet: An Extremely Efficient Convolutional Neural Network for Mobile Devices

具體見: https://github.com/kuangliu/pytorch-cifar/tree/master/models class ShuffleNet(nn.Module): def __init__(self, cfg): super(ShuffleNet, self).__init__() out_planes = cfg['out_pla
相關文章
相關標籤/搜索