search

WongKinYiu / ScaledYOLOv4

Scaled-YOLOv4: Scaling Cross Stage Partial Network
GNU General Public License v3.0
2.01k stars 570 forks source link

Freezing Layer #223

Open RaphaelWag opened 3 years ago

RaphaelWag commented 3 years ago

Does the scaled yolov4 implementation in pytorch support the option of freezing layers during training?

WongKinYiu commented 3 years ago

yes, set requires_grad=False for layers you want to freeze. reference: https://github.com/WongKinYiu/ScaledYOLOv4/blob/yolov4-large/train.py#L81-L88