From e776e8b039727dd72255ad3aa47f90cbed8890dd Mon Sep 17 00:00:00 2001 From: =?utf8?q?Fran=C3=A7ois=20Fleuret?= Date: Sun, 2 Jul 2023 22:11:12 +0200 Subject: [PATCH] Update. --- stack.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/stack.py b/stack.py index 458ec79..3844161 100755 --- a/stack.py +++ b/stack.py @@ -80,7 +80,7 @@ def seq_to_str(seq, nb_stacks, nb_digits, recorded_stack_counts=None): ###################################################################### if __name__ == "__main__": - nb, nb_steps, nb_stacks, nb_digits = 150000, 10, 1, 1 + nb, nb_steps, nb_stacks, nb_digits = 150000, 20, 2, 1 seq, recorded_stack_counts = generate_sequences( nb=nb, nb_steps=nb_steps, @@ -88,8 +88,6 @@ if __name__ == "__main__": nb_digits=nb_digits, ) - print("-- TRAIN -----------------------------") - for n in range(min(10, seq.size(0))): # print(seq_to_str(seq[n], recorded_stack_counts[n])) print(seq_to_str(seq[n], nb_stacks=nb_stacks, nb_digits=nb_digits)) -- 2.20.1