Skip to content

Commit 2dc8cbe

Browse files
committed
fix data parallel bug
1 parent 0827b4e commit 2dc8cbe

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

segmentation_module.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717

1818
def make_model(opts, classes=None):
1919
if opts.norm_act == 'iabn_sync':
20-
norm = partial(InPlaceABNSync, activation="leaky_relu", activation_param=.01)
20+
norm = partial(InPlaceABNSync, activation="leaky_relu", activation_param=.01, group=distributed.group.WORLD)
2121
elif opts.norm_act == 'iabn':
2222
norm = partial(InPlaceABN, activation="leaky_relu", activation_param=.01)
2323
elif opts.norm_act == 'abn':

0 commit comments

Comments
 (0)