How to register new models?

[lucastononrodrigues]

Hello,
I would like to register new models and run your code on them.
How exactly should I do it?
I tried to create new functions with @register_model and create new _cfgs but it didn't work out and accuses that the model is unknown.

raise RuntimeError('Unknown model (%s)' % model_name)

Thanks