replace SyncBN w/ BN

Enables training on CPU.
openvinotoolkit · Mar 18, 2022 · 93a7d9b · 93a7d9b
1 parent 0b8c805
commit 93a7d9b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/configs/ote/person-detection/person-detection-0301/model.py b/configs/ote/person-detection/person-detection-0301/model.py
@@ -6,7 +6,7 @@
         num_stages=4,
         out_indices=(0, 1, 2, 3),
         frozen_stages=1,
-        norm_cfg=dict(type='SyncBN', requires_grad=True),
+        norm_cfg=dict(type='BN', requires_grad=True),
         norm_eval=True,
         style='pytorch'),
     neck=dict(

diff --git a/configs/ote/person-detection/person-detection-0302/model.py b/configs/ote/person-detection/person-detection-0302/model.py
@@ -7,7 +7,7 @@
         num_stages=4,
         out_indices=(0, 1, 2, 3),
         frozen_stages=1,
-        norm_cfg=dict(type='SyncBN', requires_grad=True),
+        norm_cfg=dict(type='BN', requires_grad=True),
         norm_eval=True,
         style='pytorch'),
     neck=dict(