mmclassification/configs/_base_/models/shufflenet_v2_1x.py

# model settings
model = dict(
    type='ImageClassifier',
    backbone=dict(type='ShuffleNetV2', widen_factor=1.0),
    neck=dict(type='GlobalAveragePooling'),
    head=dict(
        type='LinearClsHead',
        num_classes=1000,
        in_channels=1024,
        loss=dict(type='CrossEntropyLoss', loss_weight=1.0),
        topk=(1, 5),
    ))
Add configs for MNIST, CIFAR10 and ImageNet 2020-07-09 23:59:10 +08:00			`# model settings`
			`model = dict(`
			`type='ImageClassifier',`
			`backbone=dict(type='ShuffleNetV2', widen_factor=1.0),`
			`neck=dict(type='GlobalAveragePooling'),`
			`head=dict(`
			`type='LinearClsHead',`
			`num_classes=1000,`
			`in_channels=1024,`
Update shufflenet config (#31) * update shufflenet config * remove old config Co-authored-by: lixiaojie <lixiaojie@sensetime.com> 2020-08-13 10:46:39 +08:00			`loss=dict(type='CrossEntropyLoss', loss_weight=1.0),`
			`topk=(1, 5),`
			`))`