Args: kernel_size (Tuple): kernel size of the projection layer. stride (Tuple): stride of the projection layer. padding (Tuple): padding size of the projection layer. in_chans (int): Number of input image channels. embed_dim (int):
(
self,
kernel_size: Tuple[int, ...] = (7, 7),
stride: Tuple[int, ...] = (4, 4),
padding: Tuple[int, ...] = (3, 3),
in_chans: int = 3,
embed_dim: int = 768,
)
| 68 | """ |
| 69 | |
| 70 | def __init__( |
| 71 | self, |
| 72 | kernel_size: Tuple[int, ...] = (7, 7), |
| 73 | stride: Tuple[int, ...] = (4, 4), |
| 74 | padding: Tuple[int, ...] = (3, 3), |
| 75 | in_chans: int = 3, |
| 76 | embed_dim: int = 768, |
| 77 | ): |
| 78 | """ |
| 79 | Args: |
| 80 | kernel_size (Tuple): kernel size of the projection layer. |
| 81 | stride (Tuple): stride of the projection layer. |
| 82 | padding (Tuple): padding size of the projection layer. |
| 83 | in_chans (int): Number of input image channels. |
| 84 | embed_dim (int): embed_dim (int): Patch embedding dimension. |
| 85 | """ |
| 86 | super().__init__() |
| 87 | self.proj = nn.Conv2d( |
| 88 | in_chans, embed_dim, kernel_size=kernel_size, stride=stride, padding=padding |
| 89 | ) |
| 90 | |
| 91 | def forward(self, x: torch.Tensor) -> torch.Tensor: |
| 92 | x = self.proj(x) |
nothing calls this directly
no outgoing calls
no test coverage detected