diff options
| -rw-r--r-- | model.py | 4 |
1 files changed, 3 insertions, 1 deletions
@@ -369,8 +369,10 @@ class PrimedGenerator(Runner): sub_sequence_b = out_sequences[:, i-n : i] # recursive tmp_sub_sequence = tmp_sequences[:, i-n : i] # tmp/output + pj = bottom_frame_size - n for j in range(n): - ratio = (bottom_frame_size-j-1) / (bottom_frame_size-1) + ratio = (pj + j) / (bottom_frame_size-1) + 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() |