summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorJules Laplace <julescarbon@gmail.com>2018-05-15 00:08:43 +0200
committerJules Laplace <julescarbon@gmail.com>2018-05-15 00:08:43 +0200
commit3d6f4c3f720d2f76c7f6123618448c8a1b49fbfe (patch)
tree3473d8f30a3f5702ee71eb1a0adc383762df17a9
parent34cd2ee2159ed8e02de34da7507213c58d8cd33e (diff)
lol typo
-rw-r--r--model.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/model.py b/model.py
index e8088b6..62999c9 100644
--- a/model.py
+++ b/model.py
@@ -363,7 +363,7 @@ class PrimedGenerator(Runner):
tmp_sub_sequence = tmp_sequences[:, i-n : i] # tmp/output
for j in range(n):
- ratio = i / (n-1)
+ ratio = j / (n-1)
a = sub_sequence_a[:, j].float() * (1-ratio)
b = sub_sequence_b[:, j].float() * ratio
tmp_sub_sequence[:, j] = torch.clamp(a + b, 1, q_levels-1).long()
@@ -403,7 +403,7 @@ class PrimedGenerator(Runner):
)
sub_sequence = get_sub_sequence(i, bottom_frame_size)
- # sub_sequence = sequences[:, i-bottom_frame_size : i]
+ # sub_sequence = out_sequences[:, i-bottom_frame_size : i]
prev_samples = torch.autograd.Variable(
sub_sequence,