You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
{{ message }}
This repository was archived by the owner on Apr 1, 2024. It is now read-only.
This library hints at the ability to train and apply custom models, but it's not straightforward.
From the provided train.py script, you do have the ability to train a model but integrating with the rest of the pkg doesn't seem to be supported. The script exports to onnx, which from what I can tell, you can't easily input as a model path to the supported tasks.
If I am missing some documentation or easy way of going doing this, let me know. Otherwise I hope we can consider this as a feature request.