Function ShuffleNetG2

models/shufflenet.py:98–106

(channels=3,num_classes=10, dropout=False)

def ShuffleNetG2(channels=3,num_classes=10, dropout=False):
    cfg = {
        'out_planes': [200,400,800],
        'num_blocks': [4,8,4],
        'groups': 2
    }
    # print (cfg)
    # print (channels, num_classes)
    return ShuffleNet(channels=channels,num_classes=num_classes, dropout=dropout, cfg=cfg)

def ShuffleNetG3(channels=3, num_classes=10):
    cfg = {