fix the data type in Windows

Signed-off-by: Wang Shugen <wangshugen@ict.ac.cn>
ictnlp-wshugen · Jan 9, 2019 · ffe3bcc · ffe3bcc
1 parent ae36e0b
commit ffe3bcc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/first.py b/first.py
@@ -18,7 +18,7 @@ def data_gen(V, batch, nbatches):
     Generate random data for a src-tgt copy task.
     """
     for i in range(nbatches):
-        data = torch.from_numpy(numpy.random.randint(1, V, size=(batch, 10)))
+        data = torch.from_numpy(numpy.random.randint(1, V, size=(batch, 10))).long()
         data[:, 0] = 1
         src = Variable(data, requires_grad=False)
         tgt = Variable(data, requires_grad=False)