summaryrefslogtreecommitdiff
path: root/model.py
diff options
context:
space:
mode:
authorJules Laplace <julescarbon@gmail.com>2018-05-15 00:35:49 +0200
committerJules Laplace <julescarbon@gmail.com>2018-05-15 00:35:49 +0200
commit59878fa802537b2954ab505e01a4eaec25f23ee7 (patch)
tree458198564ae0213ae997f58834cfc294b319bc2d /model.py
parent4ac07532efa4baa97ac96c29664134c3e45c4817 (diff)
lol typo
Diffstat (limited to 'model.py')
-rw-r--r--model.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/model.py b/model.py
index e7a38fb..d2a6947 100644
--- a/model.py
+++ b/model.py
@@ -370,7 +370,7 @@ class PrimedGenerator(Runner):
tmp_sub_sequence = tmp_sequences[:, i-n : i] # tmp/output
for j in range(n):
- ratio = j / (bottom_frame_size-1)
+ ratio = (bottom_frame_size-j-1) / (bottom_frame_size-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()