okayyyyyyy

1 files changed, 3 insertions, 1 deletions

diff --git a/model.py b/model.py
index 86f416e..c1f3cd3 100644
--- a/model.py
+++ b/model.py
@@ -429,13 +429,15 @@ class PrimedGenerator(Runner):
             if self.cuda:
                 prev_samples = prev_samples.cuda()
 
-            print("get upper tier conditioning..")
+            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)
             out_sequences[:, i] = sample_dist.multinomial(1).squeeze(1)
 
         torch.backends.cudnn.enabled = True