from mygpt import BracketedSequence
-# from graph import save_attention_image
-save_attention_image = None
-
######################################################################
self.save_image(
result[:96],
result_dir,
- f"world_result_{n_epoch:04d}_{model.id:02d}.png",
+ f"world_prediction_{n_epoch:04d}_{model.id:02d}.png",
logger,
)
new_quizzes,
ar_mask,
deterministic_synthesis=False,
- progress_bar_desc="new quizzes",
+ progress_bar_desc="creating quizzes",
device=self.device,
)
inverted_result,
ar_mask,
deterministic_synthesis=True,
- progress_bar_desc="solving reverse quizzes",
+ progress_bar_desc="solving reversed quizzes",
device=self.device,
)
- nb_correct += (
- (
- (new_quizzes == result).long()
- * (inverted_quizzes, inverted_result).long()
- )
- .min(dim=-1)
- .values
- )
+ nb_correct += (new_quizzes == result).long().min(dim=-1).values * (
+ inverted_quizzes == inverted_result
+ ).long().min(dim=-1).values
return new_quizzes, nb_correct