wichtounet/cifar-10

Simple C++ reader for CIFAR-10 dataset

cifar-10's Issues

Test labels = training labels?

Line 292 in the code: read_test(test_limit, dataset.training_images, dataset.training_labels, [] { return Image(3, 32, 32); });
should be : read_test(test_limit, dataset.test_images, dataset.test_labels, [] { return torch::empty({3, 32, 32}, torch::kUInt8); });

IndexError

Hey,

When trying to use the code i get a "select(): index 3 out of range for tensor of size [3, 32, 32] at dimension 0", when calling it like so: auto dataset = cifar::read_dataset_3d<std::vector, torch::Tensor, uint8_t>();. I have the dataset in a cifar-10 subfolder like the readme(installed through the bashfile), has someone else encountered this error or am i just missing something simple? When troubleshooting it seemed that the reader code got the error right away before any call?

auto dataset = cifar::read_dataset<std::vector, std::vector, uint8_t, uint8_t>();, does work