train code

[chl916185]

Thanks for your great work!
How can I use your code to train a model in dataset?
@chenguanzhou
@waka629

[sam-watts]

@chenguanzhou I'd also be interested in this if you are able to share! Thank you for the great paper and inference code

[abhishek9822]

@waka629 @chenguanzhou