Ability to define custom models

[thak123]

Hi

I am planning to implement custom models on top of transformers but not using the in built classification functions.

Is this possible using the Jiant ?

[jeswan]

That should be possible. You can define your task-specific head by adding a new head to the registry (currently WIP but you can work off the feature branch: js/feature/easy_add_model. Task head factory commit is e004af8)

It is easiest to use models that are classes/subclasses of Transformers models. Let me know if you run into issues while implementing your model.

[thak123]

Thanks for the pointer @jeswan