Splitting Pytorch Dataset Into Separate Tensor of Labels and ...
discuss.pytorch.org › t › splitting-pytorch-datasetApr 09, 2020 · Hi, I’m currently trying to train a basic CNN on the CIFAR10 dataset, which I loaded using train_dataset = torchvision.datasets.CIFAR10(DATA_PATH, train=True, transform=transform, download=True), and was able to achieve decent accuracy. However, I noticed that I was tuning the hyperparameters to the test set and seeing the response, possibly overfitting it. So, I looked into using the ...