From 6230689ade27ae954793cdd76c95d982e10fe911 Mon Sep 17 00:00:00 2001 From: =?utf8?q?Fran=C3=A7ois=20Fleuret?= Date: Fri, 7 Jul 2023 13:55:51 +0200 Subject: [PATCH] Update. --- tasks.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tasks.py b/tasks.py index 3a4a164..62a8891 100755 --- a/tasks.py +++ b/tasks.py @@ -226,7 +226,7 @@ class PicoCLVR(Task): result = self.tensorize(primer) fill = result.new_full( - result.size()[:-1] + (self.height * self.width,), self.t_nul + result.size()[:-1] + (self.height * self.width + 1,), self.t_nul ) result = torch.cat((result, fill), 1) ar_mask = (result == self.t_nul).long() -- 2.39.5