fix syncbn

bc5b5cc0 · Yuxin Wu · d89b6f07 · bc5b5cc0
Commit bc5b5cc0 authored Mar 17, 2020 by Yuxin Wu
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tensorpack/models/batch_norm.py tensorpack/models/batch_norm.py +1 -1

No files found.
--- 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