leaplabthu / rank-detr Goto Github PK

Dear rank-detr authors,

Thanks for this great work. I have a question about running your code for training.
The file "rank_transformer.py" line #279 outputs_class_tmp = self.class_embed[layer_idx](output). the self.class_embed is constantly showing None. This stops the training process.

Judging from the code, it is initialized as None at # 166, and it was not modified in the rest of the code.

Thank for pointing me to the right direction!

Firstly, congratulations on being accepted to the NIPS2023 conference!
Here are some issues I encountered while running the code:

train_net.py line 113: self.model.module.criterion.matcher.iter = self.iter
AttributeError: 'RankDETR' object has no attribute 'module'

It seems that the right code is: self.model.criterion.matcher.iter = self.iter

rank_transformer.py line 279: outputs_class_tmp = self.class_embedlayer_idx
TypeError: 'NoneType' object is not subscriptable

The reason for the error is likely to be line 167: "self.class_embed = None"

Looking forward to your reply.