(self)
| 307 | self.is_bn_merged = False |
| 308 | |
| 309 | def merge_bn(self): |
| 310 | if not self.is_bn_merged: |
| 311 | self.e_mhsa.merge_bn(self.norm1) |
| 312 | self.mlp.merge_bn(self.norm2) |
| 313 | self.is_bn_merged = True |
| 314 | |
| 315 | def forward(self, x): |
| 316 | x = self.patch_embed(x) |