(self, x)
| 403 | self.chunk.add_module('drop2', nn.Dropout(drop, inplace=True)) |
| 404 | |
| 405 | def forward(self, x): |
| 406 | |
| 407 | _, _, H, W = x.size() |
| 408 | x = einops.rearrange(x, 'b c h w -> b (h w) c') |
| 409 | x = self.chunk(x) |
| 410 | x = einops.rearrange(x, 'b (h w) c -> b c h w', h=H, w=W) |
| 411 | return x |
| 412 | |
| 413 | class LayerNormProxy(nn.Module): |
| 414 |
nothing calls this directly
no outgoing calls
no test coverage detected