diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2018-05-14 23:40:49 +0200 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2018-05-14 23:40:49 +0200 |
| commit | db041b6193eab8cb68197dcd6c385c420880da84 (patch) | |
| tree | 9aa54b8364f1384c8406c42bbc37e4df9fc1e0d9 /model.py | |
| parent | fc3877c0b8ededbc94cf583960e73bc4f8856484 (diff) | |
okayyyyyyy
Diffstat (limited to 'model.py')
| -rw-r--r-- | model.py | 6 |
1 files changed, 0 insertions, 6 deletions
@@ -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 |
