modify momentum test

5 years ago · 446828f221
parent 48735c25ca
commit 446828f221
1 changed files with 1 additions and 1 deletions
--- 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