Training code

[cxmscb]

Hi menelaoskanakis. This work is much great! Did you plan to release the training code ?

Thank you.