diff options
Diffstat (limited to 'model.py')
| -rw-r--r-- | model.py | 5 |
1 files changed, 3 insertions, 2 deletions
@@ -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, |