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