File tree Expand file tree Collapse file tree 1 file changed +10
-2
lines changed
python/paddle/fluid/transpiler Expand file tree Collapse file tree 1 file changed +10
-2
lines changed Original file line number Diff line number Diff line change @@ -187,12 +187,17 @@ def _init_splited_vars(self, slice_var_up):
187
187
188
188
param_list = []
189
189
grad_list = []
190
+ param_grad_set = set ()
190
191
for p , g in self .params_grads :
191
192
# skip parameter marked not trainable
192
193
if type (p ) == Parameter and p .trainable == False :
193
194
continue
194
- param_list .append (p )
195
- grad_list .append (g )
195
+ if p .name not in param_grad_set :
196
+ param_list .append (p )
197
+ param_grad_set .add (p .name )
198
+ if g .name not in param_grad_set :
199
+ grad_list .append (g )
200
+ param_grad_set .add (g .name )
196
201
197
202
self ._update_dist_lookup_table_vars (param_list , grad_list ,
198
203
self .params_grads )
@@ -829,6 +834,9 @@ def _create_vars_from_blocklist(self,
829
834
if not block_map .has_key (varname ):
830
835
block_map [varname ] = []
831
836
block_map [varname ].append ((long (offset ), long (size )))
837
+ # Do not remove this important debug message:
838
+ print ("block map: %s" % block_map )
839
+
832
840
for varname , splited in block_map .iteritems ():
833
841
orig_var = program .global_block ().var (varname )
834
842
if len (splited ) == 1 :
You can’t perform that action at this time.
0 commit comments