lol typo

1 files changed, 3 insertions, 3 deletions

diff --git a/model.py b/model.py
index f9c9259..e7a38fb 100644
--- a/model.py
+++ b/model.py
@@ -370,7 +370,7 @@ class PrimedGenerator(Runner):
             tmp_sub_sequence = tmp_sequences[:, i-n : i]  # tmp/output
 
             for j in range(n):
-                ratio = j / (n-1)
+                ratio = j / (bottom_frame_size-1)
                 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()
@@ -413,8 +413,8 @@ class PrimedGenerator(Runner):
                     rnn, prev_samples, upper_tier_conditioning
                 )
 
-            # sub_sequence = get_sub_sequence(i, bottom_frame_size)
-            sub_sequence = out_sequences[:, i-bottom_frame_size : i]
+            sub_sequence = get_sub_sequence(i, bottom_frame_size)
+            # sub_sequence = out_sequences[:, i-bottom_frame_size : i]
 
             prev_samples = torch.autograd.Variable(
                 sub_sequence,