Skip to content

Commit a68b3a7

Browse files
committed
get_saveable_params_list, small fix for reuse_params
1 parent 6b8ddb4 commit a68b3a7

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

TFNetwork.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -728,6 +728,8 @@ def get_saveable_params_list(self):
728728
for layer_name, layer in sorted(self.layers.items()):
729729
assert isinstance(layer, LayerBase)
730730
for param_name, param in sorted(layer.get_saveable_params_dict().items()):
731+
if param in l: # could happen with reuse_params
732+
continue
731733
l.append(param)
732734
l += self.get_auxiliary_params()
733735
l += self.extra_vars_to_save

0 commit comments

Comments
 (0)