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 @@ -186,12 +186,17 @@ def _init_splited_vars(self, split_method):
186
186
187
187
param_list = []
188
188
grad_list = []
189
+ param_grad_set = set ()
189
190
for p , g in self .params_grads :
190
191
# skip parameter marked not trainable
191
192
if type (p ) == Parameter and p .trainable == False :
192
193
continue
193
- param_list .append (p )
194
- grad_list .append (g )
194
+ if p .name not in param_grad_set :
195
+ param_list .append (p )
196
+ param_grad_set .add (p .name )
197
+ if g .name not in param_grad_set :
198
+ grad_list .append (g )
199
+ param_grad_set .add (g .name )
195
200
196
201
self ._update_dist_lookup_table_vars (param_list , grad_list ,
197
202
self .params_grads )
@@ -802,6 +807,9 @@ def _create_vars_from_blocklist(self,
802
807
if not block_map .has_key (varname ):
803
808
block_map [varname ] = []
804
809
block_map [varname ].append ((long (offset ), long (size )))
810
+ # Do not remove this important debug message:
811
+ print ("block map: %s" % block_map )
812
+
805
813
for varname , splited in block_map .iteritems ():
806
814
orig_var = program .global_block ().var (varname )
807
815
if len (splited ) == 1 :
You can’t perform that action at this time.
0 commit comments