Polish code

test=develop
6 years ago · 212242c4e4
parent 1b10a7843c
commit 212242c4e4
2 changed files with 189 additions and 192 deletions
--- a/paddle/fluid/imperative/layer.h
+++ b/paddle/fluid/imperative/layer.h
@ -243,9 +243,11 @@ class PYBIND11_HIDDEN OpBase {
    }
    // remove op desc from block desc
    if (op_desc_) {
      if (block_) {
        block_->RemoveOpInternal(op_desc_);
      }
    }
    // release resource
    for (framework::OpDesc* desc : grad_op_descs_) {
--- a/python/paddle/fluid/tests/unittests/test_imperative_basic.py
+++ b/python/paddle/fluid/tests/unittests/test_imperative_basic.py