okayyyyyyy

2 files changed, 1 insertions, 7 deletions

diff --git a/gen-prime.sh b/gen-prime.sh
index c819114..76b509e 100755
--- a/gen-prime.sh
+++ b/gen-prime.sh
@@ -54,7 +54,7 @@ function gen_prime_set () {
   # gen_prime $1 6 44100 'zero'
   # gen_prime $1 6 44100 'noise'
   # gen_prime $1 6 44100 'sin' 440
-  gen_prime $1 6 100 'noise' 0 0 True
+  gen_prime $1 6 44100 'noise' 0 0 True
   # gen_prime $1 6 44100 'sin' 440 0 True
 }
 
diff --git a/model.py b/model.py
index 06b3cfe..5b4c594 100644
--- a/model.py
+++ b/model.py
@@ -415,20 +415,14 @@ class PrimedGenerator(Runner):
             if self.cuda:
                 prev_samples = prev_samples.cuda()
 
-            print("get upper tier conditioning.. {}".format(i % bottom_frame_size))
             upper_tier_conditioning = \
                 frame_level_outputs[0][:, i % bottom_frame_size, :] \
                                       .unsqueeze(1)
-            print(upper_tier_conditioning.shape)
             sample_dist = self.model.sample_level_mlp(
                 prev_samples, upper_tier_conditioning
             ).squeeze(1).exp_().data
-            print(sample_dist.shape)
             multi = sample_dist.multinomial(1)
-            print(multi.shape)
             pred = multi.squeeze(1)
-            print(pred.shape)
-            print(out_sequences.shape)
             out_sequences[:, i] = pred
 
         torch.backends.cudnn.enabled = True