Browse Source

fix(mge/autodiff): fix incorrect handling of tuple dy

GitOrigin-RevId: beca8e3711
release-1.5
Megvii Engine Team 4 years ago
parent
commit
01354337a9
1 changed files with 1 additions and 1 deletions
  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:
dys = [ones_like(y) for y in ys]
elif isinstance(dy, (tuple, list)):
dys = ys
dys = dy
else:
dys = [dy]
try:


Loading…
Cancel
Save