- nb, seq_len, nb_stacks, nb_values = 3, 10, 1, 5
- result = generate(nb=nb, seq_len=seq_len, nb_stacks=nb_stacks, nb_values=nb_values)
- for n in range(result.size(0)):
- print(seq_to_str(result[n]))
+ nb, nb_steps, nb_stacks, nb_values = 150000, 10, 1, 5
+ seq, depth_counts = generate(
+ nb=nb, nb_steps=nb_steps, nb_stacks=nb_stacks, nb_values=nb_values
+ )
+
+ for n in range(min(10, seq.size(0))):
+ print(seq_to_str(seq[n], depth_counts[n]))