From 4ac07532efa4baa97ac96c29664134c3e45c4817 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Tue, 15 May 2018 00:31:01 +0200 Subject: lol typo --- model.py | 6 +++--- 1 file 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, -- cgit v1.2.3-70-g09d2