summaryrefslogtreecommitdiff
path: root/model.py
diff options
context:
space:
mode:
Diffstat (limited to 'model.py')
-rw-r--r--model.py6
1 files changed, 0 insertions, 6 deletions
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