hustvl / MapTR

[ICLR'23 Spotlight] MapTR: Structured Modeling and Learning for Online Vectorized HD Map Construction
MIT License
982 stars 150 forks source link

Something about layer init parameters #171

Open gaosanyuan opened 1 month ago

gaosanyuan commented 1 month ago

Could someone give me some suggestions? Why init the last layer of cls_branches like this?

bias_init = bias_init_with_prob(0.01) for m in self.clsbranches: nn.init.constant(m[-1].bias, bias_init)