whenever I\'m loading my custom dataset (reduced CIFAR-10 to 512 features) and try to train it using the common methods, I run into a ValueError. Since I\'ve been handed a H