+ 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],show_stack_nb=nb_stacks>1))
+
+ print("-- TEST ------------------------------")
+
+ remove_poped_values(seq, nb_stacks)
+