An error in the code

[LvRuH]

Hello, I found an error in the code while trying to train the model using the Visdrone dataset, in mmdet-dntr/mmdet/models/roi_heads/cascade_roi_head_cas_t2t_new_jit_mask.py line 411 ` ##### visdrone dataset ###

self.t2t_clshead = ModuleList([nn.Linear(128, 11) for in range(self.num_stages)])`

it should be ModuleList([nn.Linear(128*9, 11),or it will cause thr error: numel: integer multiplication overflow

[hoiliu-0801]

According to the Visdrone categories, the number here should be 10+1(cls token). Maybe somewhere in the codes should be adjusted.

[LvRuH]

Fine,I will check it later.Thanks for your reply！