Oh, sorry. :\
I have not completed the entir project, and after testing, there are something wrong in it, train phase on GPU is too slow, and I will upload my project to github, thanks for your attention.  In addition, Im new in this domain