- self.train_input, self.train_pred_masks = self.tensorize(train_pairs)
- self.test_input, self.test_pred_masks = self.tensorize(test_pairs)
+ self.train_input, self.train_pred_masks = self.tensorize(
+ train_pairs, shuffle=shuffle
+ )
+ self.test_input, self.test_pred_masks = self.tensorize(
+ test_pairs, shuffle=shuffle
+ )