(self, x, size=None)
| 395 | self.prenorm = prenorm |
| 396 | |
| 397 | def forward(self, x, size=None): |
| 398 | x = x + self.drop_path(self.mixer(self.norm1(x))) |
| 399 | x = x + self.drop_path(self.mlp(self.norm2(x))) |
| 400 | return x |
| 401 | |
| 402 | class OSRA_Attention(nn.Module): ### OSRA |
| 403 | def __init__(self, dim, |
nothing calls this directly
no outgoing calls
no test coverage detected