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/autodiff): fix incorrect handling of tuple dy 

GitOrigin-RevId: beca8e3711
release-1.5
Megvii Engine Team 4 years ago
parent
5868d1fe4f
commit
01354337a9
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/autodiff/grad_manager.py

+ 1
- 1
imperative/python/megengine/autodiff/grad_manager.py View File

@@ -262,7 +262,7 @@ class GradManager:
if dy is None: if dy is None:
dys = [ones_like(y) for y in ys] dys = [ones_like(y) for y in ys]
elif isinstance(dy, (tuple, list)): elif isinstance(dy, (tuple, list)):
dys = ys
dys = dy
else: else:
dys = [dy] dys = [dy]
try: try:


Write Preview
Loading…
Cancel
Save