Update shufflenet config (#31)

* update shufflenet config

* remove old config

Co-authored-by: lixiaojie <lixiaojie@sensetime.com>

configs/_base_/models/shufflenet_v1_1x.py

@@ -8,4 +8,5 @@ model = dict(
         num_classes=1000,
         in_channels=960,
         loss=dict(type='CrossEntropyLoss', loss_weight=1.0),
+        topk=(1, 5),
     ))

configs/_base_/models/shufflenet_v2_1x.py

@@ -7,4 +7,6 @@ model = dict(
         type='LinearClsHead',
         num_classes=1000,
         in_channels=1024,
-        loss=dict(type='CrossEntropyLoss', loss_weight=1.0)))
+        loss=dict(type='CrossEntropyLoss', loss_weight=1.0),
+        topk=(1, 5),
+    ))

configs/_base_/schedules/imagenet_bs1024_lineardecay.py

@@ -1,12 +0,0 @@
-# optimizer
-optimizer = dict(type='SGD', lr=0.5, momentum=0.9, weight_decay=0.00004)
-optimizer_config = dict(grad_clip=None)
-# learning policy
-lr_config = dict(
-    policy='poly',
-    min_lr=0,
-    by_epoch=False,
-    warmup='constant',
-    warmup_iters=5000,
-)
-total_epochs = 360

configs/_base_/schedules/imagenet_bs1024_linearlr_bn_nowd.py

@@ -1,5 +1,10 @@
 # optimizer
-optimizer = dict(type='SGD', lr=0.5, momentum=0.9, weight_decay=0.00004)
+optimizer = dict(
+    type='SGD',
+    lr=0.5,
+    momentum=0.9,
+    weight_decay=0.00004,
+    paramwise_cfg=dict(norm_decay_mult=0))
 optimizer_config = dict(grad_clip=None)
 # learning policy
 lr_config = dict(

configs/imagenet/shufflenet_v1_1x_batch1024_linearlr_bn_nowd.py

@@ -1,6 +1,6 @@
 _base_ = [
     '../_base_/models/shufflenet_v1_1x.py',
     '../_base_/datasets/imagenet_bs64_pil_resize.py',
-    '../_base_/schedules/imagenet_bs1024_lineardecay.py',
+    '../_base_/schedules/imagenet_bs1024_linearlr_bn_nowd.py',
     '../_base_/default_runtime.py'
 ]

configs/imagenet/shufflenet_v2_1x_batch1024_linearlr_bn_nowd.py

@@ -1,6 +1,6 @@
 _base_ = [
     '../_base_/models/shufflenet_v2_1x.py',
     '../_base_/datasets/imagenet_bs64_pil_resize.py',
-    '../_base_/schedules/imagenet_bs2048_lineardecay.py',
+    '../_base_/schedules/imagenet_bs1024_linearlr_bn_nowd.py',
     '../_base_/default_runtime.py'
 ]