Method
__init__
(
self,
in_channels: int,
out_channels: int,
kernel_size: int = 3,
stride: int = 2,
padding: int = 0,
compress_time: bool = False,
)
Source from the content-addressed store, hash-verified
| 304 | """ |
| 305 | |
| 306 | def __init__( |
| 307 | self, |
| 308 | in_channels: int, |
| 309 | out_channels: int, |
| 310 | kernel_size: int = 3, |
| 311 | stride: int = 2, |
| 312 | padding: int = 0, |
| 313 | compress_time: bool = False, |
| 314 | ): |
| 315 | super().__init__() |
| 316 | |
| 317 | self.conv = nn.Conv2d(in_channels, out_channels, kernel_size=kernel_size, stride=stride, padding=padding) |
| 318 | self.compress_time = compress_time |
| 319 | |
| 320 | def forward(self, x: torch.Tensor) -> torch.Tensor: |
| 321 | if self.compress_time: |
Callers
nothing calls this directly
Tested by
no test coverage detected