Thanks for the great work! I'm interested in the training cost of this work. What gpu need for training?