!50 fix missing batch_size

Merge pull request !50 from 吕昱峰（Nate.River）/master
mindspore-ai · Dec 27, 2021 · d25e0f3 · d25e0f3
2 parents 7810173 + dedadb7
commit d25e0f3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/chapter06/lstm/src/lstm.py b/chapter06/lstm/src/lstm.py
@@ -26,7 +26,8 @@ def __init__(self,
                  num_layers,
                  bidirectional,
                  num_classes,
-                 weight):
+                 weight,
+                 batch_size):
         super(SentimentNet, self).__init__()
         # Map words to vectors
         self.embedding = nn.Embedding(vocab_size,