diff --git a/python/paddle/v2/framework/framework.py b/python/paddle/v2/framework/framework.py index 7f5da571ba6d1c6f3570cbf5aecbcfa06d9a6862..acc61e66da1a55ec0123b195ebb52ce0858cdf5e 100644 --- a/python/paddle/v2/framework/framework.py +++ b/python/paddle/v2/framework/framework.py @@ -20,10 +20,10 @@ class Variable(object): if name is None: name = Variable._unique_var_name_() - try: - self.desc = self.block.desc.find_var(name) - is_new_var = False - except core.EnforceNotMet: + is_new_var = False + self.desc = self.block.desc.find_var(name) + + if self.desc is None: self.desc = self.block.desc.var(name) is_new_var = True diff --git a/python/paddle/v2/framework/tests/test_program.py b/python/paddle/v2/framework/tests/test_program.py index c5674382a484a91268e0139ba5588b123531210e..07473d17f76b724b35c49b1a713beeb30d251088 100644 --- a/python/paddle/v2/framework/tests/test_program.py +++ b/python/paddle/v2/framework/tests/test_program.py @@ -51,7 +51,7 @@ class TestProgram(unittest.TestCase): sum_op_desc.set_input("Y", ["b1"]) sum_op_desc.set_output("Out", ["out2"]) - target = block.new_var("out2") + target = block.var("out2") expect_ops = [ "mul", "elementwise_add", "fill_constant", "elementwise_add_grad",