add clear ops in dygraph optimizers,test=develop (#17484)

resnext-opt
Zeng Jinle 6 years ago committed by GitHub
parent ca3ba378c7
commit 65dd7ec2d6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -499,6 +499,9 @@ class Optimizer(object):
optimize_ops = self.apply_optimize(
loss, startup_program=startup_program, params_grads=params_grads)
if framework.in_dygraph_mode():
framework._dygraph_tracer()._clear_ops()
return optimize_ops, params_grads

Loading…
Cancel
Save