Question: Frozen layers?

lyuwenyu / RT-DETR

[CVPR 2024] Official RT-DETR (RTDETR paddle pytorch), Real-Time DEtection TRansformer, DETRs Beat YOLOs on Real-time Object Detection. 🔥 🔥 🔥

Apache License 2.0

2.31k stars 258 forks source link

Question: Frozen layers? #284

Open aaronrmm opened 5 months ago

aaronrmm commented 5 months ago

Can you please tell me what layers are frozen when fine-tuning? I am trying to analyze the memory requirement differences between fine-tuning and full training.

Specifically fine-tuning with

cd rtdetr_pytorch && \
python tools/train.py \
-c configs/rtdetr/rtdetr_r18vd_6x_coco.yml \
-t https://github.com/lyuwenyu/storage/releases/download/v0.1/rtdetr_r18vd_5x_coco_objects365_from_paddle.pth

lyuwenyu commented 5 months ago

full training and fine-tuning have the same training parameters.