Adapting on custom dataset

[mustafaAmplifai]

Hi team thanks for the awesome work. not sure if this question has been asked before. but I am looking for some guidance on how I can use the repo to adapt the training on custom dataset. basically I can refactor the code to include a new dataloader but wondering if there's anything specific the dataloader that I need to to care of before making it work?