From 390b95e3aabd9c660babb72a9d89dc1a99fc947b Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Mon, 14 May 2018 23:01:21 +0200 Subject: okayyyyyyy --- model.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) (limited to 'model.py') 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) -- cgit v1.2.3-70-g09d2