diff options
| -rwxr-xr-x | gen-prime.sh | 2 | ||||
| -rw-r--r-- | model.py | 6 |
2 files changed, 1 insertions, 7 deletions
diff --git a/gen-prime.sh b/gen-prime.sh index c819114..76b509e 100755 --- a/gen-prime.sh +++ b/gen-prime.sh @@ -54,7 +54,7 @@ function gen_prime_set () { # gen_prime $1 6 44100 'zero' # gen_prime $1 6 44100 'noise' # gen_prime $1 6 44100 'sin' 440 - gen_prime $1 6 100 'noise' 0 0 True + gen_prime $1 6 44100 'noise' 0 0 True # gen_prime $1 6 44100 'sin' 440 0 True } @@ -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 |