Hi, I am having difficulties in using dataloader. I want to train the network on ImageNet with a fixed number of training images. i.e. I hope to set max_images_train during the training. But I don’t know how to achieve this in pytorch. Thanks!
Are you trying to limit the number of unique images that are used? You can probably do this by editing the DatasetFolder
dataloader:
https://pytorch.org/vision/stable/_modules/torchvision/datasets/folder.html#datasetfolder
or by simply creating a modified dataset on disk with some of the images removed.
Yes, that’s what I want to do. Thank you for your reply, I will try that.