(self)
| 366 | return x, H, W |
| 367 | |
| 368 | def flops(self): |
| 369 | Ho, Wo = self.patches_resolution |
| 370 | flops = Ho * Wo * self.embed_dim * self.in_chans * (self.patch_size[0] * self.patch_size[1]) |
| 371 | if self.norm is not None: |
| 372 | flops += Ho * Wo * self.embed_dim |
| 373 | return flops |
| 374 | |
| 375 | |
| 376 | class FocalNet(nn.Module): |
nothing calls this directly
no outgoing calls
no test coverage detected