How do i train my own dataset?

my dataset has forty thousands images, but I do not know how to put my data into the neural network.I have four classes, every class in the same folder, so I have four folders .So how do I label the image?

Write a custom Dataset (http://pytorch.org/docs/0.2.0/data.html#torch.utils.data.Dataset) and use DataLoader to load it. :slight_smile:

Actually, this may also just work (and is simpler) http://pytorch.org/docs/0.2.0/torchvision/datasets.html#torchvision.datasets.ImageFolder

okay,thanks very much