This work is great and it helps me a lot! May I know the GPU memory needed to train the model and how long will it take to finish training? Thanks!