projects
/
picoclvr.git
/ blobdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
|
commitdiff
|
tree
raw
| inline |
side by side
Update.
[picoclvr.git]
/
tasks.py
diff --git
a/tasks.py
b/tasks.py
index
a97ec2e
..
889d4a9
100755
(executable)
--- 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():