From a68b3a743c2f8ae58cdd875cd9247b5784e1d93f Mon Sep 17 00:00:00 2001 From: Albert Zeyer Date: Thu, 28 Jun 2018 19:08:13 +0200 Subject: [PATCH] get_saveable_params_list, small fix for reuse_params --- TFNetwork.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/TFNetwork.py b/TFNetwork.py index 8f1ee104e3..0cefffc016 100644 --- a/TFNetwork.py +++ b/TFNetwork.py @@ -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