about train code

[CUITCHENSIYU]

Thank you for your work, it looks very interesting. I hope I can replicate this work, would it be possible to open source the related training code?

[peterdudfield]

Unfortunately we have not trained the model. But if someone wanted to add their training code we could definately incorporate it.