Dataset() object when first called with the
dset = torchvision.datasets.CIFAR10(...)
option contains the following params:
We can see that the
test_list is also pointing to the test set in our dataset, would it be valuable to have the ability to load the test set with
dset.train=False instead of having to call
torchvision.datasets.CIFAR10(..., train=False) again, or is it completely wrong what I’m proposing?
Just setting the attribute won’t change anything, since
self.target was already loaded as seen in these lines of code.
You would therefore have to implement a new method which reloads the test/train data, which would basically be another call to
In my opinion, you should definitely create separate train and test datasets, as even using separate datasets you often encounter code with data leakage.
Also the training and test transformation often differ, which you would also have to pass again to the custom method.
Thanks for the clarifications!