Method init_weights

embodiedscan/models/necks/mink_neck.py:128–131

Initialize weights.

(self)

                                                dimension=3)

    def init_weights(self):
        """Initialize weights."""
        nn.init.normal_(self.conv_cls.kernel, std=.01)
        nn.init.constant_(self.conv_cls.bias, bias_init_with_prob(.01))

    def forward(self, x: List[Tensor], batch_size) -> Tuple[List[Tensor], ...]:
        """Forward pass.