How can I train custom dataset?

[Praveenk8051]

I reviewed the paper and went through almost every part of the code. I executed the code on example datasets(stl_10 and cifar_10).

Now, I'm trying to adopt the code to my custom dataset.

It looks like we can no where to execute the training on custom dataset. The custom_dataset.py script exists but it is used for data augmentation.

Am i missing something or code is designed just to check on most common datasets?

[wvangansbeke]

Hi @Praveenk8051,

Indeed you shouldn't modify the file custom_dataset.py, but write your own dataset (e.g. data/cifar.py).
Please refer to the following issues: #8, #19, #34.

Hope this helps.

[Praveenk8051]

Thank you for your prompt response. I did not check the closed issues earlier.
Appreciate it.