WongKinYiu / PyTorch_YOLOv4

PyTorch implementation of YOLOv4
1.88k stars 585 forks source link

How to implement freeze layer #427

Open saesaria opened 1 year ago

saesaria commented 1 year ago

Hi, I have trained yolov4 for my custom dataset from scratch and then in the second try I use pre-trained yolov4.137.conv but both result are not as expected. Now I want to try to train yolov4 with freeze the backbone of the pre-trained yolov4.137.conv to gain better result.

I have tried to follow this but it didn't work. Please help how to implement it.

Thank you in advance!

AleksanderB31 commented 4 months ago

It doesnt work for me too.