Merge pull request #41 from grommy/31-fix-save-model-error

Fixed issue #31. Error on model saving
richliao · Sep 12, 2019 · 3d6f630 · 3d6f630
2 parents 93c9e2f + f3ff6c1
commit 3d6f630
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)