-
-
Notifications
You must be signed in to change notification settings - Fork 288
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
categories of pretrained weight problem #305
Comments
You don't need to do this yourself. By default, mismatched weights will be discarded. https://github.com/lyuwenyu/RT-DETR/blob/main/rtdetr_pytorch/src/solver/solver.py#L142 |
Thank you. However, I am curious if not modifying the pretrained weights will result in longer computation times. Also, since there are only 5 categories, I wonder if I need to change all instances of num_classes = 80 to 5 in the code, and modify the category items as well. (I want to minimize computation as much as possible.) |
Yes, you need to do so |
Thank you for your response. |
You can try to use ps. |
Hello, I’d like to ask where exactly the code for the CCFF module is located. I haven’t been able to find it. Thank you very much! |
I am planning to use a pretrained weight file to fine-tune my custom dataset with transfer learning. My dataset has 5 categories, whereas I anticipate that the rt-detr weight file is designed for 80 categories. Therefore, I need to modify the weight file to accommodate only 5 categories, but I am not sure how to open the weight file. I would appreciate your assistance with this.
Additionally, I am curious to know if it is necessary to modify the weight file due to the difference in the number of categories.
The text was updated successfully, but these errors were encountered: