Search policy on my custom dataset

[JavierClearImageAI]

Hi, the code seems good for research purposes. There is not much documentation, but that's ok.
I think I've managed to understand what it does and how it works.

Now I need to use this package on my new custom datasets (none of the default torchvision.datasets) for production purposes.

Any idea on how to run the search.py with custom dataset?

Thanks

[kamilkoduo]

You have to redo all the code from scratch I think

The code is good as a proof of concept but is not usable as a tool yet