From f3ff6c1d8525ae93c9285cea8cd1253dec55cd52 Mon Sep 17 00:00:00 2001 From: Mykola Kondratiev Date: Fri, 21 Jun 2019 17:00:30 +0300 Subject: [PATCH] Fixed issue #31. Error on model saving --- textClassifierHATT.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/textClassifierHATT.py b/textClassifierHATT.py index 720f9b7..b79b407 100644 --- a/textClassifierHATT.py +++ b/textClassifierHATT.py @@ -143,9 +143,9 @@ def __init__(self, attention_dim): def build(self, input_shape): assert len(input_shape) == 3 - self.W = K.variable(self.init((input_shape[-1], self.attention_dim))) - self.b = K.variable(self.init((self.attention_dim, ))) - self.u = K.variable(self.init((self.attention_dim, 1))) + self.W = K.variable(self.init((input_shape[-1], self.attention_dim)), name='W') + self.b = K.variable(self.init((self.attention_dim, )), name='b') + self.u = K.variable(self.init((self.attention_dim, 1)), name='u') self.trainable_weights = [self.W, self.b, self.u] super(AttLayer, self).build(input_shape)