From 92c05a4b88f5b0de6a84f4319bb18e2687a0fc2f Mon Sep 17 00:00:00 2001 From: =?utf8?q?Fran=C3=A7ois=20Fleuret?= Date: Thu, 20 Jul 2023 13:40:12 +0200 Subject: [PATCH] Update. --- tasks.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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(): -- 2.20.1