Method merge_bn

dmidas/backbones/next_vit.py:161–162

(self, pre_norm)

        self.drop = nn.Dropout(drop)

    def merge_bn(self, pre_norm):
        merge_pre_bn(self.conv1, pre_norm)

    def forward(self, x):
        x = self.conv1(x)