Merge pull request #70 from haojin2/fix_lstm

Fix to word_language_model/word_language_model.py for python3 compatibility
awslabs · Nov 30, 2019 · eadf998 · eadf998
2 parents ec72cb7 + 48791e2
commit eadf998
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/word_language_model/word_language_model.py b/word_language_model/word_language_model.py
@@ -91,7 +91,7 @@ def eval(data_source, ctx):
     total_L = 0.0
     ntotal = 0
     hidden_states = [
-        model.begin_state(func=mx.nd.zeros, batch_size=args.batch_size/len(ctx), ctx=ctx[i])
+        model.begin_state(func=mx.nd.zeros, batch_size=int(args.batch_size/len(ctx)), ctx=ctx[i])
         for i in range(len(ctx))
     ]
     for i in range(0, data_source.shape[0] - 1, args.bptt):