From c4f4724000922416875086df0f20f7ea44c49f12 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Mon, 14 May 2018 22:43:18 +0200 Subject: wah --- model.py | 1 + 1 file changed, 1 insertion(+) (limited to 'model.py') diff --git a/model.py b/model.py index 4b83ad7..94000f7 100644 --- a/model.py +++ b/model.py @@ -421,6 +421,7 @@ class PrimedGenerator(Runner): sample_dist = self.model.sample_level_mlp( prev_samples, upper_tier_conditioning ).squeeze(1).exp_().data + print(sample_dist) out_sequences[:, i] = sample_dist.multinomial(1).squeeze(1) torch.backends.cudnn.enabled = True -- cgit v1.2.3-70-g09d2