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 ( and use DataLoader to load it. :slight_smile:

Actually, this may also just work (and is simpler)

okay´╝îthanks very much