okayyyyyyy

1 files changed, 5 insertions, 2 deletions

diff --git a/model.py b/model.py
index 6435eb4..43c0d48 100644
--- a/model.py
+++ b/model.py
@@ -385,8 +385,8 @@ class PrimedGenerator(Runner):
         print("ENTERING LOOP....")
 
         for i in range(self.model.lookback, self.model.lookback + seq_len):
-            if (i % 100) == 0:
-                print("{}...".format(i))
+            # if (i % 100) == 0:
+            print("{}...".format(i))
 
             for (tier_index, rnn) in \
                     reversed(list(enumerate(self.model.frame_level_rnns))):
@@ -417,7 +417,9 @@ class PrimedGenerator(Runner):
                 frame_level_outputs[tier_index] = self.run_rnn(
                     rnn, prev_samples, upper_tier_conditioning
                 )
+                print("ran rnn")
 
+            print("at bottom frame")
             sub_sequence = get_sub_sequence(i, bottom_frame_size)
 
             prev_samples = torch.autograd.Variable(
@@ -426,6 +428,7 @@ class PrimedGenerator(Runner):
             )
             if self.cuda:
                 prev_samples = prev_samples.cuda()
+            print "get upp"
             upper_tier_conditioning = \
                 frame_level_outputs[0][:, i % bottom_frame_size, :] \
                                       .unsqueeze(1)