From ba99237f1ab258e8f5eac0d01719e92751c9b2a3 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Mon, 14 May 2018 23:27:16 +0200 Subject: okayyyyyyy --- model.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) (limited to 'model.py') diff --git a/model.py b/model.py index bbbd6ea..ff35baa 100644 --- a/model.py +++ b/model.py @@ -378,7 +378,7 @@ class PrimedGenerator(Runner): ratio = i / (n-1) a = sub_sequence_a[:, j].float() * (1-ratio) b = sub_sequence_b[:, j].float() * ratio - tmp_sub_sequence[:, j] = (a + b).long() + tmp_sub_sequence[:, j] = clamp(a + b, 0, q_levels).long() return tmp_sub_sequence @@ -420,7 +420,8 @@ class PrimedGenerator(Runner): print("ran rnn") print("at bottom frame") - sub_sequence = get_sub_sequence(i, bottom_frame_size) + # sub_sequence = get_sub_sequence(i, bottom_frame_size) + sub_sequence = sequences[:, i-bottom_frame_size : i] prev_samples = torch.autograd.Variable( sub_sequence, -- cgit v1.2.3-70-g09d2