pretrained models/instructions for training

[KevinGoodman]

Thanks for sharing your code!
Could you please released the pretrained models mentioned or give more detailed instructions for training as well as evaluating？

[KevinGoodman]

Also if possible, it would be nice of you can re-organize some of the code(the uses of some are not clear), I find it rather messy and confusing.