Skip to content

Commit 17a311e

Browse files
committed
fix rename de-ref again
1 parent 6720198 commit 17a311e

File tree

2 files changed

+3
-4
lines changed

2 files changed

+3
-4
lines changed

python/paddle/fluid/distribute_transpiler.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -279,7 +279,6 @@ def get_pserver_program(self, endpoint):
279279
type=v.type,
280280
dtype=v.dtype,
281281
shape=v.shape)
282-
print("create origin var: ", orig_var_name)
283282
for trainer_id in xrange(self.trainers):
284283
var = pserver_program.global_block().create_var(
285284
name="%s.trainer_%d" % (orig_var_name, trainer_id),
@@ -288,7 +287,6 @@ def get_pserver_program(self, endpoint):
288287
dtype=v.dtype,
289288
shape=v.shape)
290289
recv_inputs.append(var)
291-
print("create per trainer var: ", var.name)
292290
# step3
293291
optimize_block = pserver_program.create_block(0)
294292
# step 4

python/paddle/fluid/framework.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -773,7 +773,7 @@ def rename_var(self, name, new_name):
773773
stop_gradient = v.stop_gradient
774774
else:
775775
raise ValueError("unsupported var type: %s", type(v))
776-
776+
orig_var_type = v.type
777777
self.desc.rename_var(name, new_name)
778778
# NOTE: v is destroyed by C++ after calling rename_var.
779779
d = self.desc.find_var(new_name)
@@ -782,6 +782,7 @@ def rename_var(self, name, new_name):
782782
self,
783783
d.shape(),
784784
d.dtype(),
785+
type=orig_var_type,
785786
name=new_name,
786787
stop_gradient=stop_gradient,
787788
trainable=trainable,
@@ -792,7 +793,7 @@ def rename_var(self, name, new_name):
792793
elif var_type == "Variable":
793794
var = Variable(
794795
self,
795-
type=v.type,
796+
type=orig_var_type,
796797
name=new_name,
797798
error_clip=error_clip,
798799
stop_gradient=stop_gradient)

0 commit comments

Comments
 (0)