Conflict between the config file and model code in Deformable DETR

#3
by ZeQ - opened

Hello,
屏幕截图 2024-08-13 165926.png
I noticed a conflict in the Deformable DETR implementation. The id2label mapping in the config file does not have the last value representing the "no-object" class, which conflicts with the Deformable DETR implementation in the Hugging Face codebase. Specifically, in the modeling_deformable_detr.py file, within the DeformableDetrLoss(nn.Module) class, at line 2184, it is indicated that the "no-object" class must be the last ID (logits.shape[-1] - 1).

Thank you!

Sign up or log in to comment