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
..
0827a44
100755
(executable)
--- a/
tasks.py
+++ b/
tasks.py
@@
-34,7
+34,7
@@
def masked_inplace_autoregression(
batches,
dynamic_ncols=True,
desc=progress_bar_desc,
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():
)
with torch.autograd.no_grad():
@@
-1070,6
+1070,7
@@
class RPL(Task):
train_sequences = [
rpl.generate(
nb_starting_values=nb_starting_values,
train_sequences = [
rpl.generate(
nb_starting_values=nb_starting_values,
+ nb_result_values_max=4 * nb_starting_values,
max_input=max_input,
prog_len=prog_len,
nb_runs=nb_runs,
max_input=max_input,
prog_len=prog_len,
nb_runs=nb_runs,
@@
-1080,6
+1081,7
@@
class RPL(Task):
test_sequences = [
rpl.generate(
nb_starting_values=nb_starting_values,
test_sequences = [
rpl.generate(
nb_starting_values=nb_starting_values,
+ nb_result_values_max=4 * nb_starting_values,
max_input=max_input,
prog_len=prog_len,
nb_runs=nb_runs,
max_input=max_input,
prog_len=prog_len,
nb_runs=nb_runs,