fix syncbn

tensorpack · Mar 17, 2020 · bc5b5cc · bc5b5cc
1 parent d89b6f0
commit bc5b5cc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tensorpack/models/batch_norm.py b/tensorpack/models/batch_norm.py
@@ -333,7 +333,7 @@ def BatchNorm(inputs, axis=None, *, training=None, momentum=0.9, epsilon=1e-5,
 
         if sync_statistics is not None:
             # sync bn
-            batch_mean, batch_var = get_sync_bn_mean_var(inputs, red_axis)
+            batch_mean, batch_var = get_sync_bn_mean_var(inputs, red_axis, sync_statistics)
             batch_mean_vec = batch_mean
             batch_var_vec = batch_var