From 4cb2ed87eaa08e8e691589cc3195bfa28da727ca Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Mon, 14 May 2018 23:05:26 +0200 Subject: okayyyyyyy --- model.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) (limited to 'model.py') 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 -- cgit v1.2.3-70-g09d2