device=self.device,
)
- #!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
- self.save_quizzes(
- result_dir,
- f"DEBUG_input_{n_epoch}_{result.size(0):04d}",
- quizzes=input[:72],
- prediction=True,
- )
- self.save_quizzes(
- result_dir,
- f"DEBUG_result_{n_epoch}_{result.size(0):04d}",
- quizzes=result[:72],
- prediction=True,
- )
- #!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
-
if self.back_accuracy:
n_forward = input[:, 0] == self.token_forward
nb_total = input[n_forward].size(0)