Hello,
When I read the code I found:
self.label_enc = nn.Embedding(dn_labelbook_size + 1, hidden_dim)
In your model, the loss function is sigmoid_focal_loss, and dn_labelbook_size and num_classes are 91 when using the coco dataset.
So:
Why not use 90 but 91 for dn_labelbook_size and num_classes?
Why is the first dimension of label_enc 92 instead of 90?
Hey, the dimension of label_enc is not important, as COCO only has around 80 classes, but we use dn_labelbook_size + 1 here. You can simply ignore the empty classes that are not used.
Hello, When I read the code I found:
self.label_enc = nn.Embedding(dn_labelbook_size + 1, hidden_dim)
In your model, the loss function is sigmoid_focal_loss, and dn_labelbook_size and num_classes are 91 when using the coco dataset. So:thank you