PyTorch-Spiking-YOLOv3 icon indicating copy to clipboard operation
PyTorch-Spiking-YOLOv3 copied to clipboard

Is it possible to fine-tune using Darknet weights

Open JNaranjo-Alcazar opened this issue 3 years ago • 1 comments

Hi community, first of all, thanks for this amazing contribution.

I was wondering if it is possible to fine tune a pre-trained yolo network, for that I assume that --weights parameter should point to a downloaded file, in my case darknet53.conv.74 and then use --freze parameter and set it to True.

Would the conversion to spiking be run properly?

Thanks in advance!

JNaranjo-Alcazar avatar Sep 03 '21 06:09 JNaranjo-Alcazar

Sorry, it is impossible to fine-tune directly using Darknet weights. Because some operators(layers) can't be supported for now. You have to convert them first and train a converted pre-train model.

cwq159 avatar Sep 07 '21 03:09 cwq159