q_max and min

1 files changed, 4 insertions, 2 deletions

diff --git a/model.py b/model.py
index a6a33d5..35317f3 100644
--- a/model.py
+++ b/model.py
@@ -322,6 +322,8 @@ class PrimedGenerator(Runner):
         frame_level_outputs = [None for _ in self.model.frame_level_rnns]
 
         q_levels = self.model.q_levels
+        q_min = 0
+        q_max = q_levels
         primer_freq = 440
 
         print("_______-___-_---_-____")
@@ -333,10 +335,10 @@ class PrimedGenerator(Runner):
 
         def noise(x):
             for i in xrange(n_samples):
-                x[:, i] = random.triangular(0, q_levels)
+                x[:, i] = random.triangular(q_min, q_max)
         def sin(x):
             for i in xrange(n_samples):
-                x[:, i] = (math.sin(i/44100 * primer_freq) + 1) / 2 * q_levels
+                x[:, i] = (math.sin(i/44100 * primer_freq) + 1) / 2 * (q_max - q_min) + q_min
 
         sequences = {
           'zero': lambda x: x.fill_(utils.q_zero(self.model.q_levels)),