From a9160eaedf42c7fce906e39539c3838af8cfa405 Mon Sep 17 00:00:00 2001
From: Jules Laplace <julescarbon@gmail.com>
Date: Mon, 14 May 2018 23:37:44 +0200
Subject: okayyyyyyy

---
 model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'model.py')

diff --git a/model.py b/model.py
index 12c0e05..06b3cfe 100644
--- a/model.py
+++ b/model.py
@@ -369,7 +369,7 @@ class PrimedGenerator(Runner):
                 ratio = i / (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, 0, q_levels).long()
+                tmp_sub_sequence[:, j] = torch.clamp(a + b, 1, q_levels-1).long()
 
             return tmp_sub_sequence
 
-- 
cgit v1.2.3-70-g09d2