How to Use My Trained Yolo Model's Weights With TAO Toolkit?

Hi. I have a trained Yolov4 model with .weights and .cfg files. Can i or should i use this model with TAO?


Currently, it is not supported.
BTW, next TAO release can support PyTorch model of Unet or Classification network.
Other object detection models will be included in the future.

1 Like

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