From 496070cfe57cb41cbb3e6918b045e00ccd7c5795 Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Wed, 14 Sep 2022 15:42:20 +0800 Subject: [PATCH] fix(mge/quant): call is_distributed func correctly GitOrigin-RevId: d8ef0d62baf7603107db19d28a874acd07c2e1aa --- imperative/python/megengine/quantization/observer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/imperative/python/megengine/quantization/observer.py b/imperative/python/megengine/quantization/observer.py index f565f993..8b310586 100644 --- a/imperative/python/megengine/quantization/observer.py +++ b/imperative/python/megengine/quantization/observer.py @@ -201,7 +201,7 @@ class SyncExponentialMovingAverageObserver(ExponentialMovingAverageObserver): def forward(self, x_orig): if self.enabled: x = x_orig.detach() - if is_distributed: + if is_distributed(): min_x = all_reduce_min(x.min(), WORLD) max_x = all_reduce_max(x.max(), WORLD) else: