Remove output layer of trained Yolo

[Mshz2]

Hey there,

Thanks for the tutorial. I would like to first train the Yolo with large dataset relevant to my target class and then do transfer learning with with small dataset as the target class.

In the description, the yolov4.conv.137 which trained with coco dataset is given for doing transfer learning. Would U please help me, how can I remove the output layer of initial trained yolo of my own and get only yolo.conv weights like the given one?

Best