From 9a7ec713d189ceddedb07e9e86f152a71eec3f46 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Mon, 14 May 2018 23:29:14 +0200 Subject: okayyyyyyy --- model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'model.py') diff --git a/model.py b/model.py index ff35baa..8592ddc 100644 --- a/model.py +++ b/model.py @@ -378,7 +378,7 @@ class PrimedGenerator(Runner): ratio = i / (n-1) a = sub_sequence_a[:, j].float() * (1-ratio) b = sub_sequence_b[:, j].float() * ratio - tmp_sub_sequence[:, j] = clamp(a + b, 0, q_levels).long() + tmp_sub_sequence[:, j] = torch.clamp(a + b, 0, q_levels).long() return tmp_sub_sequence -- cgit v1.2.3-70-g09d2