diff options
Diffstat (limited to 'model.py')
| -rw-r--r-- | model.py | 8 |
1 files changed, 4 insertions, 4 deletions
@@ -388,8 +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 = out_sequences[:, i - rnn.n_frame_samples : i], + 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( @@ -414,8 +414,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, |