modify momentum test

446828f2 · changzherui · 48735c25 · 446828f2
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

tests/ut/python/ops/test_momentum.py tests/ut/python/ops/test_momentum.py +1 -1

未找到文件。
--- a/tests/ut/python/ops/test_momentum.py
+++ b/tests/ut/python/ops/test_momentum.py
@@ -38,7 +38,7 @@ def tensor_run_opt(opt, iters, learning_rate, momentum,
                   gradient, variable, moment):
    """ tensor_run_opt """
    success = True
-    new_weight = opt(variable, moment, learning_rate, gradient, momentum)
+    new_weight = opt(variable, moment, learning_rate, gradient, momentum)[0]
    success = F.depend(success, F.assign(variable, new_weight))
    return success