From: François Fleuret <francois@fleuret.org>
Date: Fri, 7 Jul 2023 11:55:51 +0000 (+0200)
Subject: Update.
X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=commitdiff_plain;h=6230689ade27ae954793cdd76c95d982e10fe911;p=picoclvr.git

Update.
---

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()