Return all labels and indexes.
(self, cuda)
| 166 | return self.get_test_set(normalization, cuda) |
| 167 | |
| 168 | def get_label_and_idxes(self, cuda): |
| 169 | """ |
| 170 | Return all labels and indexes. |
| 171 | """ |
| 172 | if cuda: |
| 173 | return self.labels_torch.cuda(), self.idx_train_torch.cuda(), self.idx_val_torch.cuda(), self.idx_test_torch.cuda() |
| 174 | return self.labels_torch, self.idx_train_torch, self.idx_val_torch, self.idx_test_torch |