Browse Source

fix(mge/quant): call is_distributed func correctly

GitOrigin-RevId: d8ef0d62ba
release-1.11.1
Megvii Engine Team 2 years ago
parent
commit
496070cfe5
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      imperative/python/megengine/quantization/observer.py

+ 1
- 1
imperative/python/megengine/quantization/observer.py View File

@@ -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:


Loading…
Cancel
Save