[Fix]Modify BN to SyncBN (open-mmlab#103)

* modify syncBN and bn_training_mode * undo modify bn_training_mode
pppppM · Mar 7, 2022 · 446621b · 446621b
1 parent a00a40b
commit 446621b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/configs/nas/detnas/detnas_evolution_search_frcnn_shufflenetv2_fpn_coco.py b/configs/nas/detnas/detnas_evolution_search_frcnn_shufflenetv2_fpn_coco.py
@@ -5,6 +5,8 @@
     workers_per_gpu=8,
 )
 
+algorithm = dict(bn_training_mode=True)
+
 searcher = dict(
     type='EvolutionSearcher',
     metrics='bbox',

diff --git a/configs/nas/detnas/detnas_supernet_frcnn_shufflenetv2_fpn_1x_coco.py b/configs/nas/detnas/detnas_supernet_frcnn_shufflenetv2_fpn_1x_coco.py
@@ -4,7 +4,7 @@
     '../../_base_/mmdet_runtime.py'
 ]
 
-norm_cfg = dict(type='BN', requires_grad=True)
+norm_cfg = dict(type='SyncBN', requires_grad=True)
 model = dict(
     type='mmdet.FasterRCNN',
     backbone=dict(