How to use one's own dataset to train the model

[yingying1012]

Thank you very much for your code! This is very useful to me! How to use one's own dataset to train the model? Looking forward to your reply.

[SongTang-x]

Thank you for your interest in our work. I don't know your dataset. I suggest you rewrite the dataloader for your dataset to adapt it to the model, and then train your model.