lol typo

1 files changed, 6 insertions, 2 deletions

diff --git a/model.py b/model.py
index f0bff1f..de46464 100644
--- a/model.py
+++ b/model.py
@@ -325,9 +325,9 @@ class PrimedGenerator(Runner):
         tmp_sequences = torch.LongTensor(n_seqs, n_samples).fill_(utils.q_zero(self.model.q_levels))
 
         q_levels = self.model.q_levels
-        q_width = q_levels >> 4
+        q_width = 64
         q_min = q_width
-        q_max = q_levels - q_width
+        q_max = q_levels - (q_width*2)
 
         print("_______-___-_---_-____")
         print("_____________--_-_-_______")
@@ -375,6 +375,10 @@ class PrimedGenerator(Runner):
                 b = sub_sequence_b[:, j].float() * ratio
                 tmp_sub_sequence[:, j] = torch.clamp(a + b, 1, q_levels-1).long()
 
+            if i == 20:
+                print(a)
+                print(b)
+                print(tmp_sub_sequence)
             return tmp_sub_sequence
 
         for i in range(self.model.lookback, self.model.lookback + seq_len):