From: François Fleuret Date: Sun, 2 Jul 2023 20:11:12 +0000 (+0200) Subject: Update. X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=commitdiff_plain;h=e776e8b039727dd72255ad3aa47f90cbed8890dd;p=picoclvr.git Update. --- 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))