(self)
| 338 | return f"input_resolution={self.input_resolution}, dim={self.dim}" |
| 339 | |
| 340 | def flops(self): |
| 341 | H, W = self.input_resolution |
| 342 | flops = H * W * self.dim |
| 343 | flops += (H // 2) * (W // 2) * 4 * self.dim * 2 * self.dim |
| 344 | return flops |
| 345 | |
| 346 | |
| 347 | class BasicLayer(nn.Module): |
nothing calls this directly
no outgoing calls
no test coverage detected