Skip to content

Commit

Permalink
get_saveable_params_list, small fix for reuse_params
Browse files Browse the repository at this point in the history
  • Loading branch information
albertz committed Jun 28, 2018
1 parent 6b8ddb4 commit a68b3a7
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions TFNetwork.py
Original file line number Diff line number Diff line change
Expand Up @@ -728,6 +728,8 @@ def get_saveable_params_list(self):
for layer_name, layer in sorted(self.layers.items()):
assert isinstance(layer, LayerBase)
for param_name, param in sorted(layer.get_saveable_params_dict().items()):
if param in l: # could happen with reuse_params
continue
l.append(param)
l += self.get_auxiliary_params()
l += self.extra_vars_to_save
Expand Down

0 comments on commit a68b3a7

Please sign in to comment.