1 files changed, 1 insertions, 0 deletions
diff --git a/model.py b/model.py
index 4b83ad7..94000f7 100644
--- a/model.py
+++ b/model.py
@@ -421,6 +421,7 @@ class PrimedGenerator(Runner):
             sample_dist = self.model.sample_level_mlp(
                 prev_samples, upper_tier_conditioning
             ).squeeze(1).exp_().data
+            print(sample_dist)
             out_sequences[:, i] = sample_dist.multinomial(1).squeeze(1)
 
         torch.backends.cudnn.enabled = True