- self.save_image(result, result_dir, f"world_result_{n_epoch:04d}.png", logger)
+ self.save_image(
+ result[:96], result_dir, f"world_result_{n_epoch:04d}.png", logger
+ )
+
+ return main_test_accuracy
+
+ def store_new_quizzes(self, new_quizzes, for_train=True):
+ input = self.train_input if for_train else self.test_input