search

WangYueFt / detr3d

MIT License
745 stars 139 forks source link

what is meaning of self.query_embedding = nn.Embedding(self.num_query, self.embed_dims * 2) #72

Open hitbuyi opened 2 months ago

hitbuyi commented 2 months ago

in detr3d_head.py, 

       if not self.as_two_stage:
            self.query_embedding = nn.Embedding(self.num_query,
                                                self.embed_dims * 2)

why self.embed_dims multiplfies *2? what's the meaning here?