training is slow

[zhiwenfan]

Hi authors, thanks for posting this great repo!
I was wondering did you notice the training speed is really slow? I use single A5000 with 24GB memory, and the overall training time could up to 70h.