convert types.. probably necessary

1 files changed, 1 insertions, 1 deletions

diff --git a/model.py b/model.py
index 53c55e8..842c727 100644
--- a/model.py
+++ b/model.py
@@ -369,7 +369,7 @@ class PrimedGenerator(Runner):
                 ratio = i / (n-1)
                 a = (1-ratio) * sub_sequence_a[:, j].float()
                 b = ratio * sub_sequence_b[:, j].float()
-                tmp_sub_sequence[:, j] = int( a + b )
+                tmp_sub_sequence[:, j] = ( a + b ).long()
 
             return tmp_sub_sequence