search

khanrc / pt.darts

PyTorch Implementation of DARTS: Differentiable Architecture Search
MIT License
439 stars 108 forks source link

How to train own dataset! #42

Open Kirito-520 opened 3 years ago

Kirito-520 commented 3 years ago

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

jacobf18 commented 3 years ago

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.