- for n in tqdm.tqdm(
- range(new_quizzes.size(0)), dynamic_ncols=True, desc="checking quizzes"
- ):
- result = new_quizzes[n][None, :].expand(nb_runs, -1).clone()
- ar_mask = (
- (torch.arange(result.size(1), device=self.device) > result.size(1) // 2)
- .long()[None, :]
- .expand_as(result)
- )