|
I want to get started on training my own model from scratch. I have already labelled all of my images so I believe as its yolox I can just use python for training. So after I have trained and exported my model can I run predictions on the model via yolox-burn or do I need another burn plugin for that? |
Replies: 1 comment
|
Looks like you've already figured it out 😄 We don't have any official training code for the YOLOX implementation, so it would require a training implementation. Or, more simply for now, just use python if that works. The trained pytorch model can be imported via any of the Once you have loaded the weights with your Burn module, you can save it in a native format (e.g., |
Looks like you've already figured it out 😄
We don't have any official training code for the YOLOX implementation, so it would require a training implementation. Or, more simply for now, just use python if that works.
The trained pytorch model can be imported via any of the
Yolox::yolox_*_pretrainedmethods which callload_weights. You only needburnandburn-storewith the features outline here to import the weights.Once you have loaded the weights with your Burn module, you can save it in a native format (e.g.,
mpk) and remove any of the import-only features.