OpenI
/
MegEngine
Not watched
1
0
Fork 0
Code
Releases 31 Wiki Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain
Browse Source

fix(mge/quant): call is_distributed func correctly 

GitOrigin-RevId: d8ef0d62ba
release-1.11.1
Megvii Engine Team 2 years ago
parent
f0291883b6
commit
496070cfe5
1 changed files with 1 additions and 1 deletions
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  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): def forward(self, x_orig):
if self.enabled: if self.enabled:
x = x_orig.detach() x = x_orig.detach()
if is_distributed:
if is_distributed():
min_x = all_reduce_min(x.min(), WORLD) min_x = all_reduce_min(x.min(), WORLD)
max_x = all_reduce_max(x.max(), WORLD) max_x = all_reduce_max(x.max(), WORLD)
else: else:


Write Preview
Loading…
Cancel
Save