- self.train_w_quizzes = sky.generate_seq(
- nb_train_samples, height=self.height, width=self.width
- ).to(device)
-
- self.test_w_quizzes = sky.generate_seq(
- nb_test_samples, height=self.height, width=self.width
- ).to(device)
+ self.train_w_quizzes = self.sky.generate_seq(nb_train_samples).to(device)
+ self.test_w_quizzes = self.sky.generate_seq(nb_test_samples).to(device)