for n in range(nb_correct.max() + 1):
recorded[n].append(new_c_quizzes[nb_correct == n].clone())
- log_string(
- f"keep c_quizzes {nb_validated()*100/nb_generated():.02f}% kept total {nb_validated()} / {nb_to_create}"
- )
+ nv = [recorded[n][-1].size(0) for n in recorded.keys()]
+
+ log_string(f"keep c_quizzes kept {nv} total {nb_validated()} / {nb_to_create}")
# concatenate and shuffle
for n in recorded.keys():