Can I use tao toolkit on my own models

May I use tao toolkit on models other than the tao pre-trained models. Like I have a trained yolov5, can I use tao toolkit to prune it, or using the QAT technology to improve it?

For yolov5, currently, it is not supported.

But for Unet or Classification, TAO provides BYOM feature to support it.
See BYOM Converter - NVIDIA Docs

How about TAO 5.0? Will be supported?

There is no update from you for a period, assuming this is not an issue anymore. Hence we are closing this topic. If need further support, please open a new one. Thanks

It is expected since TAO5.0 is planned to open sourced. And some models are not using tlt or etlt anymore.

This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.