Fixed issue richliao#31. Error on model saving

grommy · Jun 21, 2019 · f3ff6c1 · f3ff6c1
1 parent 93c9e2f
commit f3ff6c1
Showing 1 changed file with 3 additions and 3 deletions.
diff --git 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)