I reproduced the code, but I don't know how to train with my own data set, can you tell me? thank you.!!!