(self, x)
| 197 | self.is_bn_merged = True |
| 198 | |
| 199 | def forward(self, x): |
| 200 | x = self.patch_embed(x) |
| 201 | x = x + self.attention_path_dropout(self.mhca(x)) |
| 202 | if not torch.onnx.is_in_onnx_export() and not self.is_bn_merged: |
| 203 | out = self.norm(x) |
| 204 | else: |
| 205 | out = x |
| 206 | x = x + self.mlp_path_dropout(self.mlp(out)) |
| 207 | return x |
| 208 | |
| 209 | |
| 210 | class E_MHSA(nn.Module): |
nothing calls this directly
no outgoing calls
no test coverage detected