From: François Fleuret <francois@fleuret.org>
Date: Thu, 20 Jul 2023 11:40:12 +0000 (+0200)
Subject: Update.
X-Git-Url: https://www.fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=commitdiff_plain;h=92c05a4b88f5b0de6a84f4319bb18e2687a0fc2f;p=picoclvr.git

Update.
---

diff --git a/tasks.py b/tasks.py
index a97ec2e..889d4a9 100755
--- a/tasks.py
+++ b/tasks.py
@@ -34,7 +34,7 @@ def masked_inplace_autoregression(
             batches,
             dynamic_ncols=True,
             desc=progress_bar_desc,
-            # total=input.size(0) // batch_size,
+            total=(input.size(0) + batch_size - 1) // batch_size,
         )
 
     with torch.autograd.no_grad():