lol typo

1 files changed, 4 insertions, 5 deletions

diff --git a/model.py b/model.py
index 7f6be27..1f7a3fa 100644
--- a/model.py
+++ b/model.py
@@ -375,11 +375,9 @@ class PrimedGenerator(Runner):
                 ratio *= ratio
                 a = sub_sequence_a[:, j].float() * (1-ratio)
                 b = sub_sequence_b[:, j].float() * ratio
-                tmp_sub_sequence[:, j] = torch.clamp(a + b, 1, q_levels-1).long()
+                tmp_sub_sequence[:, j] = torch.clamp(a + b, 0, q_levels).long()
 
-            if i == 20:
-                print(a)
-                print(b)
+            if i == 32:
                 print(tmp_sub_sequence)
             return tmp_sub_sequence
 
@@ -390,7 +388,8 @@ class PrimedGenerator(Runner):
                 if i % rnn.n_frame_samples != 0:
                     continue
 
-                sub_sequence = get_sub_sequence(i, rnn.n_frame_samples)
+                # sub_sequence = get_sub_sequence(i, rnn.n_frame_samples)
+                sub_sequence = out_sequences[:, i - rnn.n_frame_samples : i],
 
                 prev_samples = torch.autograd.Variable(
                     2 * utils.linear_dequantize(