Skip to content
This repository was archived by the owner on Apr 17, 2023. It is now read-only.

Commit 9b8a33a

Browse files
author
Vladimir
committed
replace SyncBN w/ BN
Enables training on CPU.
1 parent 7893b80 commit 9b8a33a

File tree

2 files changed

+2
-2
lines changed
  • configs/ote/person-detection

2 files changed

+2
-2
lines changed

configs/ote/person-detection/person-detection-0301/model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
num_stages=4,
77
out_indices=(0, 1, 2, 3),
88
frozen_stages=1,
9-
norm_cfg=dict(type='SyncBN', requires_grad=True),
9+
norm_cfg=dict(type='BN', requires_grad=True),
1010
norm_eval=True,
1111
style='pytorch'),
1212
neck=dict(

configs/ote/person-detection/person-detection-0302/model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77
num_stages=4,
88
out_indices=(0, 1, 2, 3),
99
frozen_stages=1,
10-
norm_cfg=dict(type='SyncBN', requires_grad=True),
10+
norm_cfg=dict(type='BN', requires_grad=True),
1111
norm_eval=True,
1212
style='pytorch'),
1313
neck=dict(

0 commit comments

Comments
 (0)