From 24d49ab58c245182bcad0e50949045ae47903362 Mon Sep 17 00:00:00 2001
From: Yao Zou <yao_zou@hotmail.com>
Date: Fri, 20 Apr 2018 13:05:00 +0800
Subject: [PATCH] fix 32, 64-bit issue on Windows

---
 .gitignore          | 2 ++
 transformer/Beam.py | 2 +-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/.gitignore b/.gitignore
index 4986def..c37007c 100644
--- a/.gitignore
+++ b/.gitignore
@@ -101,3 +101,5 @@ ENV/
 
 # mypy
 .mypy_cache/
+
+.idea/
diff --git a/transformer/Beam.py b/transformer/Beam.py
index e86129b..ae376f2 100644
--- a/transformer/Beam.py
+++ b/transformer/Beam.py
@@ -87,7 +87,7 @@ def get_tentative_hypothesis(self):
             _, keys = self.sort_scores()
             hyps = [self.get_hypothesis(k) for k in keys]
             hyps = [[Constants.BOS] + h for h in hyps]
-            dec_seq = torch.from_numpy(np.array(hyps))
+            dec_seq = torch.from_numpy(np.array(hyps, dtype=np.int64))
 
         return dec_seq