AILab-CVC / YOLO-World

[CVPR 2024] Real-Time Open-Vocabulary Object Detection
https://www.yoloworld.cc
GNU General Public License v3.0
3.88k stars 372 forks source link

In fine tuning how to store the embeddings into the weights of classification layer #393

Open getroy74 opened 2 weeks ago

getroy74 commented 2 weeks ago

In paper, in additional details chapter A.2. Fine-tuning Details, it is mentioned that after fine tuning, text embedding can be stored in the classification layer. Which part of the code is doing this. How we can do it if in code base it is not mentioned.

Thanks