(self)
| 522 | return x |
| 523 | |
| 524 | def flops(self): |
| 525 | flops = 0 |
| 526 | flops += self.patch_embed.flops() |
| 527 | for i, layer in enumerate(self.layers): |
| 528 | flops += layer.flops() |
| 529 | flops += self.num_features * self.patches_resolution[0] * self.patches_resolution[1] // (2 ** self.num_layers) |
| 530 | flops += self.num_features * self.num_classes |
| 531 | return flops |
| 532 |