This website works better with JavaScript.
Home
Issues
Pull Requests
Milestones
Repositories
Datasets
Forum
实训
竞赛
大数据
应用
Register
Sign In
OpenI
/
MegEngine
Not watched
Unwatch
Watch all
Watch but not notify
1
Star
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
Split View
Diff Options
Show Stats
Download Patch File
Download Diff File
+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:
Write
Preview
Loading…
Cancel
Save