How to train own dataset!

[Kirito-520]

If i use this method to train own dataset, how do i change the code!

[jacobf18]

I have not tried it yet, and will update this after I've tried it, but the approach that I will take is to convert my data into a classification dataset in the format of the ImageFolder data loader class from here: https://pytorch.org/vision/stable/datasets.html. Then, I'll change some of the lines in the get_data function in utils.py to load in my data loader class instead of one of the standard ones.