| 43 | |
| 44 | |
| 45 | class Transpose(nn.Module): |
| 46 | def __init__(self, dim0, dim1): |
| 47 | super(Transpose, self).__init__() |
| 48 | self.dim0 = dim0 |
| 49 | self.dim1 = dim1 |
| 50 | |
| 51 | def forward(self, x): |
| 52 | x = x.transpose(self.dim0, self.dim1) |
| 53 | return x |
| 54 | |
| 55 | |
| 56 | def forward_vit(pretrained, x): |
no outgoing calls
no test coverage detected