thanks for your great job! I want to try, but there need access rights for the pretrained model. by the way, how to train?