From 0b2ae54eb620b3f70e082cb46ffade8571184d04 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Tue, 15 May 2018 00:25:23 +0200 Subject: lol typo --- model.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/model.py b/model.py index f0bff1f..de46464 100644 --- a/model.py +++ b/model.py @@ -325,9 +325,9 @@ class PrimedGenerator(Runner): tmp_sequences = torch.LongTensor(n_seqs, n_samples).fill_(utils.q_zero(self.model.q_levels)) q_levels = self.model.q_levels - q_width = q_levels >> 4 + q_width = 64 q_min = q_width - q_max = q_levels - q_width + q_max = q_levels - (q_width*2) print("_______-___-_---_-____") print("_____________--_-_-_______") @@ -375,6 +375,10 @@ class PrimedGenerator(Runner): b = sub_sequence_b[:, j].float() * ratio tmp_sub_sequence[:, j] = torch.clamp(a + b, 1, q_levels-1).long() + if i == 20: + print(a) + print(b) + print(tmp_sub_sequence) return tmp_sub_sequence for i in range(self.model.lookback, self.model.lookback + seq_len): -- cgit v1.2.3-70-g09d2