1 files changed, 4 insertions, 5 deletions
diff --git a/model.py b/model.py
index aa52088..4b83ad7 100644
--- a/model.py
+++ b/model.py
@@ -367,15 +367,14 @@ class PrimedGenerator(Runner):
 
             for j in range(n):
                 ratio = i / (n-1)
-                for k in range(n_seqs):
-                    a = (1-ratio) * sub_sequence_a[k, j]
-                    b = ratio * sub_sequence_b[k, j]
-                    tmp_sub_sequence[k, j] = a + b
+                a = sub_sequence_a[:, j].float() * (1-ratio)
+                b = sub_sequence_b[:, j].float() * ratio
+                tmp_sub_sequence[:, j] = (a + b).long()
 
             return tmp_sub_sequence
 
         for i in range(self.model.lookback, self.model.lookback + seq_len):
-            if i > 0 and (i % 1000) == 0:
+            if (i % 1000) == 0:
                 print("{}...".format(i))
 
             for (tier_index, rnn) in \