(self, x: torch.Tensor)
| 200 | self.resblocks = nn.Sequential(*[ResidualAttentionBlock(width, heads, attn_mask) for _ in range(layers)]) |
| 201 | |
| 202 | def forward(self, x: torch.Tensor): |
| 203 | return self.resblocks(x) |
| 204 | |
| 205 | |
| 206 | class VisionTransformer(nn.Module): |
nothing calls this directly
no outgoing calls
no test coverage detected