From 543fb3466b75f24bf79ac5658607574fc9b4402e Mon Sep 17 00:00:00 2001
From: Jules Laplace <julescarbon@gmail.com>
Date: Tue, 15 May 2018 00:37:52 +0200
Subject: lol typo

---
 model.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

(limited to 'model.py')

diff --git a/model.py b/model.py
index d2a6947..7f6be27 100644
--- a/model.py
+++ b/model.py
@@ -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()
-- 
cgit v1.2.3-70-g09d2