(self, x: torch.Tensor)
| 206 | self.resblocks = nn.Sequential(*[ResidualAttentionBlock(width, heads, attn_mask) for _ in range(layers)]) |
| 207 | |
| 208 | def forward(self, x: torch.Tensor): |
| 209 | return self.resblocks(x) |
| 210 | |
| 211 | |
| 212 | class VisualTransformer(nn.Module): |
nothing calls this directly
no outgoing calls
no test coverage detected