From 3c2b37c46da522e1ac67a8a5f2c1b8fea5985b33 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Tue, 15 May 2018 00:43:06 +0200 Subject: lol typo --- model.py | 9 ++++----- 1 file 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( -- cgit v1.2.3-70-g09d2