Add training code for the models on the model zoo

[Warvito]

For now, we added just the inference code for some models in the model zoo. It would be great to have the reference to train these models as a bundle.