From 56e70edc461e7ede5d4f27853df0940324822551 Mon Sep 17 00:00:00 2001 From: iperov Date: Thu, 19 Aug 2021 23:18:04 +0400 Subject: [PATCH] fix error in model saving --- core/leras/layers/Saveable.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/core/leras/layers/Saveable.py b/core/leras/layers/Saveable.py index 0cc0e94..e72f594 100644 --- a/core/leras/layers/Saveable.py +++ b/core/leras/layers/Saveable.py @@ -46,7 +46,9 @@ class Saveable(): raise Exception("name must be defined.") name = self.name - for w, w_val in zip(weights, nn.tf_sess.run (weights)): + + for w in weights: + w_val = nn.tf_sess.run (w).copy() w_name_split = w.name.split('/', 1) if name != w_name_split[0]: raise Exception("weight first name != Saveable.name") @@ -97,10 +99,10 @@ class Saveable(): nn.batch_set_value(tuples) except: return False - + return True def init_weights(self): nn.init_weights(self.get_weights()) - + nn.Saveable = Saveable