Training on the custom dataset

[WAL-l]

Can I use this code to train the model on my own dataset? If so, how should the dataset be structured?

[Subalzero]

You have to create a custom dataset loader in pytorch: https://pytorch.org/tutorials/beginner/basics/data_tutorial.html