Great model!Inspired me a lot. How many Gpus did you use to train this model, and how much memory did you need?