(self, x)
| 120 | ) |
| 121 | |
| 122 | def forward(self, x): |
| 123 | x = self.pooler(x).squeeze(1) |
| 124 | x = self.probe(x) |
| 125 | return x |
| 126 | |
| 127 | class AttnHead(nn.Module): |
| 128 | def __init__(self, embed_dim, num_heads, mlp_ratio, depth, norm_layer, init_std, qkv_bias, complete_block, classes=3): |
nothing calls this directly
no outgoing calls
no test coverage detected