提交 496070cf 编写于 作者: M Megvii Engine Team

fix(mge/quant): call is_distributed func correctly

GitOrigin-RevId: d8ef0d62baf7603107db19d28a874acd07c2e1aa
上级 f0291883
......@@ -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:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册