+ 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]
+
+ new_c_quizzes = torch.cat(
+ [recorded[n] for n in range(args.min_to_validate, args.max_to_validate + 1)],
+ dim=0,
+ )
+
+ new_c_quizzes = new_c_quizzes[
+ torch.randperm(new_c_quizzes.size(0), device=new_c_quizzes.device)[
+ : nb_for_train + nb_for_test
+ ]
+ ]