diff options
| -rw-r--r-- | model.py | 4 |
1 files changed, 2 insertions, 2 deletions
@@ -363,7 +363,7 @@ class PrimedGenerator(Runner): tmp_sub_sequence = tmp_sequences[:, i-n : i] # tmp/output for j in range(n): - ratio = i / (n-1) + ratio = j / (n-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() @@ -403,7 +403,7 @@ class PrimedGenerator(Runner): ) sub_sequence = get_sub_sequence(i, bottom_frame_size) - # sub_sequence = sequences[:, i-bottom_frame_size : i] + # sub_sequence = out_sequences[:, i-bottom_frame_size : i] prev_samples = torch.autograd.Variable( sub_sequence, |