About training on custom datasets

[972821054]

Hi, thank you very much for your contribution! Could you please share the code used to generate DHA_data? I'd like to reproduce your code, but I'm stuck here. I would be very grateful if you could!