(self, x)
| 187 | ])) |
| 188 | |
| 189 | def forward(self, x): |
| 190 | x = self.patch_partition(x) |
| 191 | for regular_block, shifted_block in self.layers: |
| 192 | x = regular_block(x) |
| 193 | x = shifted_block(x) |
| 194 | return x.permute(0, 3, 1, 2) |
| 195 | |
| 196 | |
| 197 | class SwinTransformer(nn.Module): |
nothing calls this directly
no outgoing calls
no test coverage detected