candle icon indicating copy to clipboard operation
candle copied to clipboard

finetuning yolo 8 candle model

Open flutter-painter opened this issue 7 months ago • 0 comments

What is the correct way to finetune yolo8 model to be used here ? Finetuning model using candle is not straightforward. candle\candle-examples\examples\yolo-v8\main.rs // model model architecture points at ultralytics : https://github.com/ultralytics/ultralytics/issues/189 But my model trained using ultralytics and converted to safetensors yield tensor errors when used in candle ylo 8 example. Renaming the tensors to match the candle yolo model did not work.

I see DarkNet struct in the model.rs so I wonder if one should rather use Darknet instead (@LaurentMazare ) ?

flutter-painter avatar May 05 '25 15:05 flutter-painter