Merge pull request #7438 from takagi/fix-example-sentiment-recursive-…

…minibatch Support default dtype in sentiment example's recursive minibatch version
chainer · Jun 25, 2019 · 5289f67 · 5289f67
2 parents e48db81 + 926480d
commit 5289f67
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/sentiment/train_recursive_minibatch.py b/examples/sentiment/train_recursive_minibatch.py
@@ -136,8 +136,8 @@ def forward(self, *inputs):
         count = 0
         correct = 0
 
-        stack = self.xp.zeros(
-            (batch, maxlen * 2, self.n_units), self.xp.float32)
+        dtype = chainer.get_dtype()
+        stack = self.xp.zeros((batch, maxlen * 2, self.n_units), dtype)
         for i, (word, label) in enumerate(zip(sequences, leaf_labels)):
             batch = word.shape[0]
             es = self.leaf(word)