+ for n in range(nb_correct.max() + 1):
+ recorded[n].append(new_c_quizzes[nb_correct == n].clone())
+
+ 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():
+ if len(recorded[n]) > 0:
+ q = torch.cat(recorded[n], dim=0)
+ q = q[torch.randperm(q.size(0), device=q.device)]
+ recorded[n] = q
+ else:
+ del recorded[n]