nb_test_samples, height=self.height, width=self.width
).to(device)
- # print()
- # for a in world.seq2str(self.train_input):
- # print(a)
- # for a in world.seq2str(self.test_input):
- # print(a)
- # exit(0)
-
self.nb_codes = max(self.train_input.max(), self.test_input.max()) + 1
self.train_quizzes = []