We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 0827b4e commit 2dc8cbeCopy full SHA for 2dc8cbe
segmentation_module.py
@@ -17,7 +17,7 @@
17
18
def make_model(opts, classes=None):
19
if opts.norm_act == 'iabn_sync':
20
- norm = partial(InPlaceABNSync, activation="leaky_relu", activation_param=.01)
+ norm = partial(InPlaceABNSync, activation="leaky_relu", activation_param=.01, group=distributed.group.WORLD)
21
elif opts.norm_act == 'iabn':
22
norm = partial(InPlaceABN, activation="leaky_relu", activation_param=.01)
23
elif opts.norm_act == 'abn':
0 commit comments