(self)
| 70 | return x |
| 71 | |
| 72 | def flops(self) -> float: |
| 73 | Ho, Wo = self.patches_resolution |
| 74 | flops = Ho * Wo * self.embed_dim * self.in_chans * (self.patch_size[0] * self.patch_size[1]) |
| 75 | if self.norm is not None: |
| 76 | flops += Ho * Wo * self.embed_dim |
| 77 | return flops |
| 78 | |
| 79 | def reset_parameters(self): |
| 80 | k = 1 / (self.in_chans * (self.patch_size[0] ** 2)) |
nothing calls this directly
no outgoing calls
no test coverage detected