- while nb_validated() < nb_for_train + nb_for_test:
- nb_to_validate = nb_for_train + nb_for_test
-
- if len(model_indexes) == 0:
- model_indexes = [i.item() for i in torch.randperm(len(models))]
-
- model = models[model_indexes.pop()]
-
- new_c_quizzes, nb_correct, ave_seq_logproba = quizz_machine.create_c_quizzes(
- nb=nb_to_validate,
- model_for_generation=model,
- models_for_validation=models,
+ nb_to_create = nb_for_train + nb_for_test
+
+ while nb_validated() < nb_to_create:
+ (
+ new_c_quizzes,
+ nb_correct,
+ ave_seq_logproba,
+ ) = quizz_machine.gang_create_c_quizzes(
+ nb=nb_to_create,
+ nb_models_for_generation=args.nb_models_for_generation,
+ models=models,
+ mode=args.generation_mode,